Update app.py

app.py

@@ -61,7 +61,10 @@ async def send_chat_message(LLM_URL, LLM_API, user_input):
         "conversation_id": "",
         "user": USER_ID,
     }
-    print("Sending chat message payload:", payload)  # Debug information
+    print("Sending chat message payload:", payload)
+
+    full_response = []
+    buffer = b""  # 使用字節緩衝區來處理不完整的行
 
     async with aiohttp.ClientSession() as session:
         try:
@@ -72,29 +75,47 @@ async def send_chat_message(LLM_URL, LLM_API, user_input):
                 timeout=aiohttp.ClientTimeout(total=60)
             ) as response:
                 if response.status != 200:
-                    print(f"Error: {response.status}")
+                    error_text = await response.text()
+                    print(f"Error: {response.status}, Body: {error_text}")
                     return f"Error: {response.status}"
 
-                full_response = []
-                async for line in response.content:
-                    line = line.decode('utf-8').strip()
-                    if not line:
-                        continue
-                    if "data: " not in line:
-                        continue
-                    try:
-                        print("Received line:", line)  # Debug information
-                        data = json.loads(line.split("data: ")[1])
-                        if "answer" in data:
-                            full_response.append(data["answer"])
-                    except (IndexError, json.JSONDecodeError) as e:
-                        print(f"Error parsing line: {line}, error: {e}")  # Debug information
-                        continue
+                # 手動迭代數據塊 (chunk)
+                async for chunk in response.content.iter_any():
+                    buffer += chunk
+                    # 嘗試按行分割緩衝區中的數據
+                    while b'\n' in buffer:
+                        line, buffer = buffer.split(b'\n', 1)
+                        line_str = line.decode('utf-8').strip()
+
+                        if not line_str or "data: " not in line_str:
+                            continue
+                        
+                        try:
+                            # 檢查是否是我們要的答案，而不是工作流日誌
+                            # 這些日誌訊息 (e.g., event: "node_started") 就是造成問題的元兇
+                            # 我們只關心包含 "answer" 的 message_chunk
+                            if '"event": "message_chunk"' in line_str or '"answer"' in line_str:
+                                print("Received line:", line_str) # 只打印有用的行
+                                data_part = line_str.split("data: ", 1)[1]
+                                data = json.loads(data_part)
+                                
+                                if "answer" in data:
+                                    full_response.append(data["answer"])
+                                # 如果是 Dify API，答案可能在 data['data']['answer']
+                                elif "data" in data and "answer" in data["data"]:
+                                    full_response.append(data["data"]["answer"])
+
+                        except (IndexError, json.JSONDecodeError) as e:
+                            # 忽略無法解析的行，它們很可能是日誌
+                            # print(f"Skipping unparsable line: {line_str}, error: {e}")
+                            pass
 
                 if full_response:
                     return ''.join(full_response).strip()
                 else:
-                    return "Error: No response found in the response"
+                    # 如果循環結束都沒有收到答案，返回提示
+                    return "抱歉，我無法從 API 取得有效的回覆。"
+                    
         except Exception as e:
             print(f"Exception: {e}")
             return f"Exception: {e}"