Update app.py

app.py

@@ -120,30 +120,31 @@ class PollinationsModel:
             }
             response = requests.post(self.api_url, json=payload, timeout=120)
             response.raise_for_status()
-            result = response.json()
-            if "content" in result:
-                return result["content"]
-            else:
-                logger.error(f"Unexpected response structure: {result}")
-                return "Error: Unexpected API response format"
+            # Die API gibt direkt Text zurück, kein JSON
+            return response.text
         except Exception as e:
             logger.exception(f"PollinationsModel generate failed: {e}")
             return f"Error generating response: {str(e)}"
             
-    def __call__(self, prompt=None, messages=None, **kwargs):
+    def __call__(self, messages=None, **kwargs):
         """Make the model callable as required by smolagents CodeAgent."""
-        # Determine content to send based on what's provided
-        if messages:
-            # If we have chat messages, extract content from the last one
-            content = messages[-1].get("content", "") if messages else ""
-        else:
-            content = prompt or ""
+        prompt = ""
+        if messages and isinstance(messages, list) and len(messages) > 0:
+            # Get content from the last message
+            if hasattr(messages[-1], 'content'):
+                prompt = messages[-1].content
+            elif isinstance(messages[-1], dict) and 'content' in messages[-1]:
+                prompt = messages[-1]['content']
         
-        # Generate the response
-        response_text = self.generate(content, **kwargs)
+        response_text = self.generate(prompt)
         
-        # Return in a format similar to OpenAI API responses
-        return {"choices": [{"message": {"content": response_text}}]}
+        # Create a response object with the content attribute
+        class ChatMessage:
+            def __init__(self, content):
+                self.content = content
+                
+        return ChatMessage(response_text)
+
 
 
 logger.info(f"Initializing Pollinations LLM connection: {MODEL_ID}")