Update app.py

app.py

@@ -12,27 +12,32 @@ taskgen_client = Client("tencent/Hunyuan-Large")
 OUTPUT_DIR = "outputs"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
-# 拆解 JD 成任务，并生成三个解决方案（增强容错）
-def generate_solutions_from_jd(jd):
-    message = f"""你是一个岗位分析助手，请根据以下JD内容：\n
-1. 首先识别一个可以用来测试岗位候选人核心能力的具体任务；
-2. 然后为这个任务生成三个不同的解决方案建议。
-
-请严格以如下格式输出：
-任务:（任务描述）\n方案1:（内容）\n方案2:（内容）\n方案3:（内容）\n
-JD: {jd}
-"""
-    response = taskgen_client.predict(message=message, api_name="/chat")
+# 拆解 JD 成任务
+def extract_task_from_jd(jd):
+    message = f"""你是一个岗位分析助手，请根据以下JD内容提取一个可以用来测试候选人核心能力的具体任务。
+请以如下格式回复：
+任务:（任务描述）
 
+JD: {jd}"""
+    response = taskgen_client.predict(message=message, api_name="/chat")
     task_match = re.search(r"任务[:：](.*)", response)
-    solutions = re.findall(r"方案[123][:：](.*)", response)
+    return task_match.group(1).strip() if task_match else "任务解析失败"
 
-    task = task_match.group(1).strip() if task_match else "任务解析失败"
+# 基于任务生成三个解决方案
+def generate_solutions_from_task(task):
+    message = f"""你是一个解决方案生成助手，请根据以下任务设计三种不同的实现思路。
+请严格按如下格式输出：
+方案1:（内容）
+方案2:（内容）
+方案3:（内容）
+
+任务: {task}"""
+    response = taskgen_client.predict(message=message, api_name="/chat")
+    solutions = re.findall(r"方案[123][:：](.*)", response)
     s1 = solutions[0].strip() if len(solutions) > 0 else "方案1解析失败"
     s2 = solutions[1].strip() if len(solutions) > 1 else "方案2解析失败"
     s3 = solutions[2].strip() if len(solutions) > 2 else "方案3解析失败"
-
-    return task, s1, s2, s3
+    return s1, s2, s3
 
 # 构建 Gradio UI
 def build_ui():
@@ -41,9 +46,9 @@ def build_ui():
 
         jd_input = gr.Textbox(label="输入 JD", placeholder="请输入岗位描述 JD")
         task_output = gr.Textbox(label="拆解出的测试任务", interactive=False)
+        generate_task_btn = gr.Button("🧠 拆解 JD 成任务")
 
-        generate_button = gr.Button("🚀 拆解任务并生成方案")
-
+        generate_solutions_btn = gr.Button("🚀 基于任务生成三个方案")
         sol1 = gr.Textbox(label="方案1", interactive=False)
         sol2 = gr.Textbox(label="方案2", interactive=False)
         sol3 = gr.Textbox(label="方案3", interactive=False)
@@ -57,9 +62,13 @@ def build_ui():
 
         task_state = gr.State()
 
-        def handle_jd_input(jd_text):
-            task, s1, s2, s3 = generate_solutions_from_jd(jd_text)
-            return task, s1, s2, s3, task
+        def handle_task_gen(jd_text):
+            task = extract_task_from_jd(jd_text)
+            return task, task
+
+        def handle_solutions_gen(task_text):
+            s1, s2, s3 = generate_solutions_from_task(task_text)
+            return s1, s2, s3
 
         def handle_submit(selected_idx, user_input_text, comment_text, task_text):
             record = {
@@ -77,12 +86,12 @@ def build_ui():
             except Exception as e:
                 return f"❌ 保存失败：{str(e)}"
 
-        generate_button.click(fn=handle_jd_input, inputs=[jd_input], outputs=[task_output, sol1, sol2, sol3, task_state])
+        generate_task_btn.click(fn=handle_task_gen, inputs=[jd_input], outputs=[task_output, task_state])
+        generate_solutions_btn.click(fn=handle_solutions_gen, inputs=[task_state], outputs=[sol1, sol2, sol3])
         submit.click(fn=handle_submit, inputs=[select_radio, user_solution, comment, task_state], outputs=[feedback])
 
     return demo
 
 if __name__ == "__main__":
     demo = build_ui()
-    demo.launch()
-
+    demo.launch()