Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 import requests
 import pandas as pd
 from smolagents import CodeAgent, DuckDuckGoSearchTool, tool
+from smolagents.models import BaseModel
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
 logger = logging.getLogger(__name__)
@@ -97,7 +98,8 @@ Formatting Rules for FINAL ANSWER:
 Let's begin!
 """
 
-class PollinationsModel:
+# Benutzerdefiniertes Modell für Pollinations API
+class PollinationsModel(BaseModel):
     def __init__(self, model_id="openai-large", max_tokens=8196, seed=42):
         self.model_id = model_id
         self.max_tokens = max_tokens
@@ -120,32 +122,15 @@ class PollinationsModel:
             }
             response = requests.post(self.api_url, json=payload, timeout=120)
             response.raise_for_status()
-            # Die API gibt direkt Text zurück, kein JSON
-            return response.text
+            result = response.json()
+            if "content" in result:
+                return result["content"]
+            else:
+                logger.error(f"Unexpected response structure: {result}")
+                return "Error: Unexpected API response format"
         except Exception as e:
             logger.exception(f"PollinationsModel generate failed: {e}")
             return f"Error generating response: {str(e)}"
-            
-    def __call__(self, messages=None, **kwargs):
-        """Make the model callable as required by smolagents CodeAgent."""
-        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']
-        
-        response_text = self.generate(prompt)
-        
-        # 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}")
 try:
@@ -249,4 +234,4 @@ logger.info("Gradio interface setup complete.")
 if __name__ == "__main__":
     logger.info("Launching Gradio application...")
     demo.launch(debug=True, share=False)
-    logger.info("Gradio application launched.")
+    logger.info("Gradio application launched.")