Update app.py

app.py

@@ -38,14 +38,19 @@ def predict(image_input, question):
     # 修正：使用 text 參數而不是直接傳遞 messages
     result = pipe(text=messages)
     
-    # 修正輸出格式：直接提取 assistant 的回應
-    if isinstance(result, list) and len(result) >= 2:
-        # 通常第二個元素是 assistant 的回應
-        assistant_message = result[1]
-        if isinstance(assistant_message, dict) and assistant_message.get("role") == "assistant":
-            return assistant_message.get("content", "無法獲取回應")
+    # 修正輸出格式：處理新的結果格式
+    # 檢查是否是新格式：包含 generated_text 的 dict
+    if isinstance(result, list) and len(result) > 0 and isinstance(result[0], dict):
+        first_result = result[0]
+        if "generated_text" in first_result:
+            generated_text = first_result["generated_text"]
+            # generated_text 是一個 list，找到 assistant 的回應
+            if isinstance(generated_text, list):
+                for message in generated_text:
+                    if isinstance(message, dict) and message.get("role") == "assistant":
+                        return message.get("content", "無法獲取回應")
     
-    # 備用：遍歷找到 assistant
+    # 備用：舊格式處理
     if isinstance(result, list):
         for message in result:
             if isinstance(message, dict) and message.get("role") == "assistant":