Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import pandas as pd | |
| from llm import OpenAI3, OpenAI4 | |
| from question import generate | |
| block_css = """ | |
| .importantButton { | |
| background: linear-gradient(45deg, #7e0570,#5d1c99, #6e00ff) !important; | |
| border: none !important; | |
| } | |
| .importantButton:hover { | |
| background: linear-gradient(45deg, #ff00e0,#8500ff, #6e00ff) !important; | |
| border: none !important; | |
| }""" | |
| webui_title = "📚📚📚📚📚📚📚📚📚📚📚📚* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * ☘️ * * *智能生题* * * ☘️ * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *📚📚📚📚📚📚📚📚📚📚📚📚" | |
| llm_name_dict = {'chatgpt-3.5': 'OpenAI3', 'chatgpt-4': 'OpenAI4', 'chatglm—6b': 'ChatGLM'} | |
| kg = pd.read_excel('教材知识点.xlsx') | |
| qt = pd.read_excel('题型.xlsx') | |
| def llm_change(name): | |
| llm = eval(eval('llm_name_dict[name]'))() | |
| return llm | |
| def clear_screen(chatbot): | |
| return [[None, None]] | |
| def function_select(mode): | |
| if mode == "试题生成": | |
| return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) | |
| elif mode == "试卷生成": | |
| return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False) | |
| elif mode == "阅读理解生成": | |
| return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False) | |
| else: | |
| return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True) | |
| def update_semesters(_publisher): | |
| df = kg[kg['教材'] == _publisher] | |
| _semesters = list(set(df['学期'])) | |
| _semesters.sort() | |
| return gr.inputs.Dropdown.update(choices=_semesters) | |
| def update_units(_publisher, _semester): | |
| df = kg[kg['教材'] == _publisher] | |
| df = df[df['学期'] == _semester] | |
| _units = list(set(df['进度'])) | |
| _units.sort() | |
| return gr.inputs.Dropdown.update(choices=_units) | |
| def update_material_and_skill(_publisher, _semester, _unit, _material_type): | |
| df = kg[kg['教材'] == _publisher] | |
| df = df[df['学期'] == _semester] | |
| df = df[df['进度'] == _unit] | |
| _materials = df[_material_type].to_list() | |
| _materials = list(set(_materials[0].split('\n'))) | |
| _materials.sort() | |
| if '' in _materials: | |
| _materials.pop(_materials.index('')) | |
| # _materials = ['全选', '随机'] + _materials | |
| df = qt[qt['材料类型'] == _material_type] | |
| _skills = list(set(df['能指'])) | |
| _skills.sort() | |
| return gr.inputs.Radio.update(choices=_materials, value=None), gr.inputs.Radio.update(choices=_skills, value=None),\ | |
| gr.inputs.Radio.update(choices=[], value=None), gr.inputs.Radio.update(choices=[], value=None), \ | |
| gr.inputs.Textbox.update(value='') | |
| def update_question_class(_material_type, _skill): | |
| df = qt[qt['材料类型'] == _material_type] | |
| df = df[df['能指'] == _skill] | |
| _question_class = list(set(df['一级题型'])) | |
| _question_class.sort() | |
| return gr.inputs.Dropdown.update(choices=_question_class, value=None), \ | |
| gr.inputs.Dropdown.update(choices=[], value=None), \ | |
| gr.inputs.Textbox.update(value=''), | |
| def update_question_type(_material_type, _skill, _question_class): | |
| df = qt[qt['材料类型'] == _material_type] | |
| df = df[df['能指'] == _skill] | |
| df = df[df['一级题型'] == _question_class] | |
| _question_type = list(set(df['二级题型'])) | |
| _question_type.sort() | |
| return gr.inputs.Dropdown.update(choices=_question_type) | |
| def update_question_example(_question_type): | |
| if _question_type: | |
| df = qt[qt['二级题型'] == _question_type] | |
| return df['样例'].to_list()[0] | |
| else: | |
| return '' | |
| def generate_questions(_model_mode, _material, _question_type, _question_num, _chatbot): | |
| questions = generate(materials=_material, q_type=_question_type, q_num=_question_num, model=_model_mode) | |
| response = _question_type + '\n\n' | |
| for i in range(len(questions)): | |
| q = questions[i] | |
| response = response + f'{i+1}_question:\n\n' + q['question'] + '\n\n' + f'{i+1}_answer:\n\n' +\ | |
| q['answer'] + '\n\n\n\n' | |
| _chatbot = _chatbot + [[None, response]] | |
| return _chatbot | |
| with gr.Blocks(css=block_css) as demo: | |
| gr.Markdown('\n\n\n\n') | |
| gr.Markdown(webui_title) | |
| gr.Markdown('\n\n\n\n') | |
| llm = gr.State('') | |
| # model_mode = gr.Radio(['chatgpt-3.5', "chatgpt-4", 'chatglm—6b'], label="请选择驱动模型") | |
| model_mode = gr.Radio(['chatgpt-3.5'], label="请选择驱动模型") | |
| model_mode.change(fn=llm_change, inputs=[model_mode], outputs=[llm]) | |
| # fun_mode = gr.Radio(["试题生成"], label="请选择功能模式", value="试题生成") | |
| # fun_mode = gr.Radio(["试题生成","试卷生成", "阅读理解生成", "作文批改"], label="请选择功能模式", value="试卷生成") | |
| qg = gr.Column(visible=True) | |
| pg = gr.Column(visible=False) | |
| rc = gr.Row(visible=False) | |
| aa = gr.Row(visible=False) | |
| # fun_mode.change(fn=function_select, inputs=[fun_mode], outputs=[qg, pg, rc, aa]) | |
| # progress.change(fn=llm_change, inputs=[model_mode], outputs=[llm]) | |
| with qg: | |
| with gr.Row(): | |
| with gr.Column(scale=10): | |
| publishers = list(set(kg['教材'])) | |
| publishers.sort() | |
| publisher = gr.inputs.Dropdown(label="版本", choices=publishers) | |
| semester = gr.inputs.Dropdown(label="学期", choices=[]) | |
| unit = gr.inputs.Dropdown(label="单元", choices=[]) | |
| material_type = gr.Radio(['词汇', '短语', '句子', '语法'], label="出题材料类型") | |
| material = gr.CheckboxGroup([], label="出题材料") | |
| skill = gr.Radio([], label="考察能力") | |
| question_class = gr.Radio([], label="题类") | |
| question_type = gr.Radio(label="题型", choices=[]) | |
| question_example = gr.Textbox('', label='题例') | |
| publisher.change(update_semesters, inputs=[publisher], outputs=[semester]) | |
| semester.change(update_units, inputs=[publisher, semester], outputs=[unit]) | |
| skill.change(update_question_class, inputs=[material_type, skill], | |
| outputs=[question_class, question_type, question_example]) | |
| question_class.change(update_question_type, inputs=[material_type, skill, question_class], | |
| outputs=[question_type]) | |
| question_type.change(update_question_example, inputs=[question_type], outputs=[question_example]) | |
| material_type.change(update_material_and_skill, inputs=[publisher, semester, unit, material_type], | |
| outputs=[material, skill, question_class, question_type, question_example]) | |
| # option_num = gr.Slider(0, 5, value=3, step=1, label="选项数量", interactive=True) | |
| question_num = gr.Slider(0, 5, value=1, step=1, label="题量", interactive=True) | |
| # question = gr.Textbox('', label='生成题目') | |
| generate_button = gr.Button('生成') | |
| with gr.Column(scale=10): | |
| chatbot = gr.Chatbot([[None, None]], | |
| elem_id="chat-box", | |
| show_label=False).style(height=1000) | |
| generate_button.click(generate_questions, inputs=[model_mode, material, question_type, question_num, | |
| chatbot], outputs=chatbot) | |
| # with pg: | |
| # | |
| # publisher = gr.inputs.Dropdown(label="教材版本", choices=textbooks) | |
| # semester = gr.inputs.Dropdown(label="教材版本", choices=['三年级上', '三年级下']) | |
| # progress = gr.Radio(['unit1', 'unit2', 'unit3', 'unit4', 'unit5', 'unit6', 'unit7', '期中', 'unit8', 'unit9', | |
| # 'unit10', 'unit11', 'unit12', 'unit13', 'unit14', '期末'], label="请选择教学进度") | |
| # | |
| # with gr.Row(): | |
| # | |
| # with gr.Column(scale=10): | |
| # | |
| # gr.Markdown('\n\n\n\n') | |
| # gr.Markdown('📚**************word pronunciation**************📚') | |
| # gr.Markdown('\n\n\n\n') | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('choice') | |
| # gr.Markdown('\n') | |
| # | |
| # word_p1 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="选择划线部分发音不同的单词", | |
| # interactive=True) | |
| # | |
| # word_p2 = gr.Slider(0, | |
| # 5, | |
| # value=1, | |
| # step=1, | |
| # label="选择与指定单词划线部分发音不同的单词", | |
| # interactive=True) | |
| # | |
| # word_p3 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="以下划线字母哪个发/r/的音", | |
| # interactive=True) | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('judge') | |
| # gr.Markdown('\n') | |
| # | |
| # word_p4 = gr.Slider(0, | |
| # 5, | |
| # value=1, | |
| # step=1, | |
| # label="判断下列单词划线部分读音是否相同", | |
| # interactive=True) | |
| # gr.Markdown('\n\n\n\n') | |
| # gr.Markdown('📚**************word spelling**************📚') | |
| # gr.Markdown('\n\n\n\n') | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('choice') | |
| # gr.Markdown('\n') | |
| # | |
| # word_s1 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="选出拼写正确的单词", | |
| # interactive=True) | |
| # | |
| # word_s2 = gr.Slider(0, | |
| # 5, | |
| # value=1, | |
| # step=1, | |
| # label="选择正确的拼写补全单词", | |
| # interactive=True) | |
| # | |
| # word_s3 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="看图，选出该图表示单词的首字母", | |
| # interactive=True) | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('fill') | |
| # gr.Markdown('\n') | |
| # | |
| # word_s4 = gr.Slider(0, | |
| # 5, | |
| # value=1, | |
| # step=1, | |
| # label="根据图片补全单词", | |
| # interactive=True) | |
| # gr.Markdown('\n\n\n\n') | |
| # gr.Markdown('📚**************word meaning**************📚') | |
| # gr.Markdown('\n\n\n\n') | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('choice') | |
| # gr.Markdown('\n') | |
| # | |
| # word_m1 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="挑选一组词中不同类的词", | |
| # interactive=True) | |
| # | |
| # word_m2 = gr.Slider(0, | |
| # 5, | |
| # value=1, | |
| # step=1, | |
| # label="挑选单词的正确释义", | |
| # interactive=True) | |
| # | |
| # word_m3 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="选择可以替换划线部分的词", | |
| # interactive=True) | |
| # word_m4 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="选择与图片相符的单词", | |
| # interactive=True) | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('judge') | |
| # gr.Markdown('\n') | |
| # | |
| # word_m5 = gr.Slider(0, | |
| # 5, | |
| # value=1, | |
| # step=1, | |
| # label="判断单词中文意思是否正确", | |
| # interactive=True) | |
| # | |
| # word_m6 = gr.Slider(0, | |
| # 5, | |
| # value=1, | |
| # step=1, | |
| # label="判断下列单词是否属于同一类", | |
| # interactive=True) | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('fill') | |
| # gr.Markdown('\n') | |
| # | |
| # word_m7 = gr.Slider(0, | |
| # 5, | |
| # value=1, | |
| # step=1, | |
| # label="根据汉语提示补全句子", | |
| # interactive=True) | |
| # | |
| # word_m8 = gr.Slider(0, | |
| # 5, | |
| # value=1, | |
| # step=1, | |
| # label="根据图片提示补全句子", | |
| # interactive=True) | |
| # | |
| # word_m9 = gr.Slider(0, | |
| # 5, | |
| # value=1, | |
| # step=1, | |
| # label="下厨下列单词的反义词", | |
| # interactive=True) | |
| # | |
| # gr.Markdown('\n\n\n\n') | |
| # gr.Markdown('📚**************fixed phrase**************📚') | |
| # gr.Markdown('\n\n\n\n') | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('choice') | |
| # gr.Markdown('\n') | |
| # | |
| # phrase_f1 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="请选择合适的词语将句子补充完整", | |
| # interactive=True) | |
| # | |
| # gr.Markdown('\n\n\n\n') | |
| # gr.Markdown('📚**************sentence structure**************📚') | |
| # gr.Markdown('\n\n\n\n') | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('fill') | |
| # gr.Markdown('\n') | |
| # | |
| # sentence_s1 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="请将下列单词组成句子", | |
| # interactive=True) | |
| # | |
| # gr.Markdown('\n\n\n\n') | |
| # gr.Markdown('📚**************sentence meaning**************📚') | |
| # gr.Markdown('\n\n\n\n') | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('fill') | |
| # gr.Markdown('\n') | |
| # | |
| # sentence_s2 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="请将下列汉语翻译成对应英语", | |
| # interactive=True) | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('choice') | |
| # gr.Markdown('\n') | |
| # | |
| # sentence_s3 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="选出正确的中文翻译", | |
| # interactive=True) | |
| # | |
| # | |
| # gr.Markdown('\n\n\n\n') | |
| # gr.Markdown('📚**************名词所有格**************📚') | |
| # gr.Markdown('\n\n\n\n') | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('choice') | |
| # gr.Markdown('\n') | |
| # | |
| # grammar_1 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="请选择正确的单词进行填空", | |
| # interactive=True) | |
| # | |
| # gr.Markdown('\n\n\n\n') | |
| # gr.Markdown('📚**************there be 句型**************📚') | |
| # gr.Markdown('\n\n\n\n') | |
| # | |
| # gr.Markdown('\n') | |
| # gr.Markdown('choice') | |
| # gr.Markdown('\n') | |
| # | |
| # grammar_2 = gr.Slider(0, | |
| # 5, | |
| # value=2, | |
| # step=1, | |
| # label="请选择正确的单词进行填空", | |
| # interactive=True) | |
| # | |
| # # gr.Markdown(f'试卷存储路径') | |
| # # paper_path = gr.Textbox(show_label=False, placeholder="请输入待生成试卷存放路径", | |
| # # value='.\\papers\\', lines=1).style(container=False) | |
| # | |
| # generate_button = gr.Button(value="生成试卷", visible=False) | |
| # | |
| # with gr.Column(scale=10): | |
| # | |
| # chatbot = gr.Chatbot([[None, None]], | |
| # elem_id="chat-box", | |
| # show_label=False).style(height=1000) | |
| # | |
| # generate_button.click(fn=generate_paper, | |
| # inputs=[progress, word_p1, word_p2, word_p3, word_p4, word_s1, word_s2, word_s3, word_s4, | |
| # word_m1, word_m2, word_m3, word_m4, word_m5, word_m6, word_m7, word_m8, word_m9, | |
| # phrase_f1, sentence_s1, sentence_s2, sentence_s3, grammar_1, grammar_2], | |
| # outputs=[chatbot]) | |
| # with rc: | |
| # | |
| # with gr.Column(scale=10): | |
| # chatbot = gr.Chatbot([[None, None]], | |
| # elem_id="chat-box", | |
| # show_label=False).style(height=800) | |
| # clear_button = gr.Button(value="清屏") | |
| # clear_button.click(fn=clear_screen, inputs=[chatbot], outputs=[chatbot]) | |
| # | |
| # with gr.Column(scale=10): | |
| # | |
| # with gr.Tab('生成配置'): | |
| # generate_number = gr.Slider(1, | |
| # 5, | |
| # value=3, | |
| # step=1, | |
| # label="请设定单篇文章需要生成的问题数量", | |
| # interactive=True) | |
| # | |
| # gr.Markdown(f'单篇生成') | |
| # texts = gr.Textbox(show_label=False, placeholder="文本内容", lines=12).style(container=False) | |
| # text_button = gr.Button(value="生成问题") | |
| # text_button.click(fn=text_generate, inputs=[chatbot, texts, generate_number, llm], outputs=[chatbot]) | |
| # | |
| # gr.Markdown(f'批量生成') | |
| # folder_address = gr.File(label="添加文件", | |
| # file_types=['.txt', '.md', '.docx', '.pdf'], | |
| # file_count="multiple", | |
| # show_label=False | |
| # ) | |
| # file_button = gr.Button(value="生成问题") | |
| # file_button.click(fn=files_generate, inputs=[chatbot, folder_address, generate_number, llm], | |
| # outputs=[chatbot]) | |
| # | |
| # with aa: | |
| # | |
| # with gr.Column(scale=10): | |
| # chatbot = gr.Chatbot([[None, None]], | |
| # elem_id="chat-box", | |
| # show_label=False).style(height=800) | |
| # clear_button = gr.Button(value="清屏") | |
| # clear_button.click(fn=clear_screen, inputs=[chatbot], outputs=[chatbot]) | |
| # | |
| # with gr.Column(scale=10): | |
| # | |
| # with gr.Tab('批改配置'): | |
| # | |
| # gr.Markdown(f'写作要求') | |
| # require_texts = gr.Textbox(show_label=False, placeholder="文本内容", lines=4).style(container=False) | |
| # # require_texts_button = gr.Button(value="批改") | |
| # # require_texts_button.click(fn=text_correct, inputs=[chatbot, texts, llm], outputs=[chatbot]) | |
| # | |
| # gr.Markdown(f'单篇批改') | |
| # feedback_texts = gr.Textbox(show_label=False, placeholder="文本内容", lines=12).style(container=False) | |
| # feedback_button = gr.Button(value="批改") | |
| # feedback_button.click(fn=text_correct, inputs=[chatbot, require_texts, feedback_texts, llm], | |
| # outputs=[chatbot]) | |
| # | |
| # gr.Markdown(f'批量批改') | |
| # folder_address = gr.File(label="添加文件", | |
| # file_types=['.txt', '.md', '.docx', '.pdf'], | |
| # file_count="multiple", | |
| # show_label=False | |
| # ) | |
| # file_button = gr.Button(value="批改") | |
| # file_button.click(fn=files_correct, inputs=[chatbot, require_texts, folder_address, llm], | |
| # outputs=[chatbot]) | |
| # demo.queue(concurrency_count=5).launch(server_name='0.0.0.0', server_port=7858, share=True) | |
| demo.queue(concurrency_count=5).launch() | |