Update app.py

app.py

@@ -3,7 +3,6 @@ import os
 from datetime import datetime
 from gradio_client import Client
 import re
-import uuid
 import json
 
 # 初始化任务生成客户端（腾讯混元 Space）
@@ -57,10 +56,19 @@ def generate_solutions_from_task(task):
 
 任务: {task}"""
     response = taskgen_client.predict(message=message, api_name="/chat")
-    # fallback：若正则提取失败，直接展示原始响应
-    solutions = re.findall(r"方案[123][:：]\s*(.*)", response)
+    
+    # 输出原始响应内容以便调试
+    print("原始响应内容：", response)
+    
+    # 尝试通过正则提取所有方案内容
+    solutions = re.findall(r"方案[123][:：]\s*(.*?)(?=\n|方案[123][:：]|$)", response, re.DOTALL)
+    
+    # 输出提取的方案内容，方便调试
+    print("提取的方案内容：", solutions)
+    
     if len(solutions) < 3:
         return response.strip(), "（解析失败，显示原始回复）", ""
+    
     s1 = solutions[0].strip()
     s2 = solutions[1].strip()
     s3 = solutions[2].strip()