Update app.py

app.py

@@ -20,8 +20,7 @@ def extract_task_from_jd(jd):
 
 JD: {jd}"""
     response = taskgen_client.predict(message=message, api_name="/chat")
-    task_match = re.search(r"任务[:：](.*)", response)
-    return task_match.group(1).strip() if task_match else "任务解析失败"
+    return response.strip()  # 直接返回原始响应内容，便于调试
 
 # 基于任务生成三个解决方案
 def generate_solutions_from_task(task):
@@ -33,10 +32,13 @@ def generate_solutions_from_task(task):
 
 任务: {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解析失败"
+    # fallback：若正则提取失败，直接展示原始响应
+    solutions = re.findall(r"方案[123][:：]\s*(.*)", response)
+    if len(solutions) < 3:
+        return response.strip(), "（解析失败，显示原始回复）", ""
+    s1 = solutions[0].strip()
+    s2 = solutions[1].strip()
+    s3 = solutions[2].strip()
     return s1, s2, s3
 
 # 构建 Gradio UI
@@ -45,13 +47,13 @@ def build_ui():
         gr.Markdown("## 📌 JD 任务拆解 + 解决方案选择 Demo")
 
         jd_input = gr.Textbox(label="输入 JD", placeholder="请输入岗位描述 JD")
-        task_output = gr.Textbox(label="拆解出的测试任务", interactive=False)
+        task_output = gr.Textbox(label="拆解出的测试任务", lines=2, interactive=False)
         generate_task_btn = gr.Button("🧠 拆解 JD 成任务")
 
         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)
+        sol1 = gr.Textbox(label="方案1 / 或原始回复", lines=4, interactive=False)
+        sol2 = gr.Textbox(label="方案2", lines=4, interactive=False)
+        sol3 = gr.Textbox(label="方案3", lines=4, interactive=False)
 
         select_radio = gr.Radio(choices=["1", "2", "3"], label="请选择你最满意的解决方案编号")
         user_solution = gr.Textbox(lines=4, label="📄 或者填写你自己的解决方案（可选）")