Update app.py

app.py

@@ -12,7 +12,7 @@ taskgen_client = Client("tencent/Hunyuan-Large")
 OUTPUT_DIR = "outputs"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
-# 拆解 JD 成任务，并生成三个解决方案
+# 拆解 JD 成任务，并生成三个解决方案（增强容错）
 def generate_solutions_from_jd(jd):
     message = f"""你是一个岗位分析助手，请根据以下JD内容：\n
 1. 首先识别一个可以用来测试岗位候选人核心能力的具体任务；
@@ -25,14 +25,12 @@ JD: {jd}
     response = taskgen_client.predict(message=message, api_name="/chat")
 
     task_match = re.search(r"任务[:：](.*)", response)
-    solution1 = re.search(r"方案1[:：](.*)", response)
-    solution2 = re.search(r"方案2[:：](.*)", response)
-    solution3 = re.search(r"方案3[:：](.*)", response)
+    solutions = re.findall(r"方案[123][:：](.*)", response)
 
     task = task_match.group(1).strip() if task_match else "任务解析失败"
-    s1 = solution1.group(1).strip() if solution1 else "方案1解析失败"
-    s2 = solution2.group(1).strip() if solution2 else "方案2解析失败"
-    s3 = solution3.group(1).strip() if solution3 else "方案3解析失败"
+    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
 
@@ -86,4 +84,5 @@ def build_ui():
 
 if __name__ == "__main__":
     demo = build_ui()
-    demo.launch()
+    demo.launch()
+