Yaoliang commited on
Commit
cf6c1df
·
1 Parent(s): b40b8b9

fix conflict and continue rebase

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. 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"————【{file_name}】————")
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"🚀 问题{idx}{q}\n💡 答案{idx}{a}\n")
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()