Spaces:
Sleeping
Sleeping
fix conflict and continue rebase
Browse files- app.py +3 -3
- document_processor.py +15 -0
app.py
CHANGED
|
@@ -62,15 +62,15 @@ def format_qa_output_per_file(results):
|
|
| 62 |
"""
|
| 63 |
output = []
|
| 64 |
for file_name, qa_list in results:
|
| 65 |
-
output.append(f"
|
| 66 |
if isinstance(qa_list, list):
|
| 67 |
for idx, qa in enumerate(qa_list, 1):
|
| 68 |
q = qa.get('question') or qa.get('问题') or ''
|
| 69 |
a = qa.get('answer') or qa.get('答案') or ''
|
| 70 |
-
output.append(f"
|
| 71 |
else:
|
| 72 |
output.append(str(qa_list))
|
| 73 |
-
output.append("
|
| 74 |
output.append("祝你高分通关,未来可期!⚡")
|
| 75 |
return '\n'.join(output)
|
| 76 |
|
|
|
|
| 62 |
"""
|
| 63 |
output = []
|
| 64 |
for file_name, qa_list in results:
|
| 65 |
+
output.append(f"\n**【{file_name}】AI智能考题与答案**\n")
|
| 66 |
if isinstance(qa_list, list):
|
| 67 |
for idx, qa in enumerate(qa_list, 1):
|
| 68 |
q = qa.get('question') or qa.get('问题') or ''
|
| 69 |
a = qa.get('answer') or qa.get('答案') or ''
|
| 70 |
+
output.append(f"Q{idx}: {q}\nA{idx}: {a}\n")
|
| 71 |
else:
|
| 72 |
output.append(str(qa_list))
|
| 73 |
+
output.append("\n---\n")
|
| 74 |
output.append("祝你高分通关,未来可期!⚡")
|
| 75 |
return '\n'.join(output)
|
| 76 |
|
document_processor.py
CHANGED
|
@@ -9,6 +9,8 @@ from langchain.schema import Document as LangchainDocument
|
|
| 9 |
import logging
|
| 10 |
import gradio as gr
|
| 11 |
from simple_qa import call_llm_api
|
|
|
|
|
|
|
| 12 |
|
| 13 |
logging.basicConfig(level=logging.INFO)
|
| 14 |
logger = logging.getLogger(__name__)
|
|
@@ -130,6 +132,7 @@ def qa_func(text, model_type="deepseek", max_questions=5):
|
|
| 130 |
return result
|
| 131 |
|
| 132 |
iface = gr.Interface(
|
|
|
|
| 133 |
fn=qa_func,
|
| 134 |
inputs=[
|
| 135 |
gr.Textbox(label="请输入你的文档内容或问题"),
|
|
@@ -140,6 +143,18 @@ iface = gr.Interface(
|
|
| 140 |
title="逢考必过·AI考试复习助手",
|
| 141 |
description="输入文档内容,自动生成高质量问答对,支持多模型切换。"
|
| 142 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 143 |
|
| 144 |
if __name__ == "__main__":
|
| 145 |
iface.launch()
|
|
|
|
| 9 |
import logging
|
| 10 |
import gradio as gr
|
| 11 |
from simple_qa import call_llm_api
|
| 12 |
+
from dotenv import load_dotenv
|
| 13 |
+
load_dotenv()
|
| 14 |
|
| 15 |
logging.basicConfig(level=logging.INFO)
|
| 16 |
logger = logging.getLogger(__name__)
|
|
|
|
| 132 |
return result
|
| 133 |
|
| 134 |
iface = gr.Interface(
|
| 135 |
+
<<<<<<< HEAD
|
| 136 |
fn=qa_func,
|
| 137 |
inputs=[
|
| 138 |
gr.Textbox(label="请输入你的文档内容或问题"),
|
|
|
|
| 143 |
title="逢考必过·AI考试复习助手",
|
| 144 |
description="输入文档内容,自动生成高质量问答对,支持多模型切换。"
|
| 145 |
)
|
| 146 |
+
=======
|
| 147 |
+
fn=qa_func,
|
| 148 |
+
inputs=[
|
| 149 |
+
gr.Textbox(label="请输入你的文档内容或问题"),
|
| 150 |
+
gr.Radio(["deepseek", "stepfun", "gemini"], label="选择模型"),
|
| 151 |
+
gr.Slider(1, 10, value=5, label="生成问答对数量")
|
| 152 |
+
],
|
| 153 |
+
outputs="text",
|
| 154 |
+
title="逢考必过·AI考试复习助手",
|
| 155 |
+
description="输入文档内容,自动生成高质量问答对,支持多模型切换。"
|
| 156 |
+
)
|
| 157 |
+
>>>>>>> 7cec768 (add gemini option)
|
| 158 |
|
| 159 |
if __name__ == "__main__":
|
| 160 |
iface.launch()
|