Update app.py

app.py

@@ -62,25 +62,53 @@ def infer(image: Image.Image, question: str) -> str:
 
     # ---- Robust readout (handles nested lists/dicts/strings) ----
     def extract_text(obj):
+        """
+        Normalizes pipeline outputs to a plain string.
+        Handles:
+          - {'generated_text': '...'}
+          - {'generated_text': [{'role': 'assistant', 'content': '...'}, ...]}
+          - [{'generated_text': '...'}], nested lists, or raw strings
+        """
         if obj is None:
             return ""
+    
+        # If it's already a string
         if isinstance(obj, str):
             return obj
+    
+        # If it's a dict
         if isinstance(obj, dict):
-            # Most common contract
-            if "generated_text" in obj and isinstance(obj["generated_text"], str):
-                return obj["generated_text"]
-            # Some models return {"text": "..."}
+            gen = obj.get("generated_text")
+            # Case A: direct string
+            if isinstance(gen, str):
+                return gen
+            # Case B: list of chat turns -> find the assistant message
+            if isinstance(gen, (list, tuple)) and gen:
+                for turn in reversed(gen):
+                    if isinstance(turn, dict) and turn.get("role") == "assistant":
+                        content = turn.get("content")
+                        # Some models return list[str] or str here; normalize
+                        if isinstance(content, list):
+                            return " ".join(map(str, content))
+                        return str(content) if content is not None else ""
+                # Fallback: stringify first element
+                return extract_text(gen[0])
+    
+            # Other keys sometimes used by models
             if "text" in obj and isinstance(obj["text"], str):
                 return obj["text"]
-            # Fallback: stringify dict
+    
+            # Last resort for dicts
             return str(obj)
+    
+        # If it's a list/tuple, drill down
         if isinstance(obj, (list, tuple)) and obj:
-            # Flatten by drilling down to the first non-list item
             return extract_text(obj[0])
-        # Last resort
+    
+        # Fallback
         return str(obj)
 
+
     return extract_text(out).strip()