import gradio as gr
from llm import OpenAI3, OpenAI4
from app_com.generate_paper import generate_paper
from app_com.reading_comprehension import text_generate, files_generate
from app_com.correct_assay import text_correct, files_correct
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'}
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)
elif mode == "阅读理解生成":
return 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=True)
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"], label="请选择驱动模型")
model_mode.change(fn=llm_change, inputs=[model_mode], outputs=[llm])
fun_mode = gr.Radio(["试卷生成"], label="请选择功能模式", value="试卷生成")
# fun_mode = gr.Radio(["试卷生成", "阅读理解生成", "作文批改"], label="请选择功能模式", value="试卷生成")
pg = gr.Column(visible=True)
qg = gr.Row(visible=False)
aa = gr.Row(visible=False)
fun_mode.change(fn=function_select, inputs=[fun_mode], outputs=[pg, qg, aa])
# progress.change(fn=llm_change, inputs=[model_mode], outputs=[llm])
with pg:
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 qg:
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()