update app

app.py

@@ -145,20 +145,27 @@ def run_model_inference(formatted_messages, max_tokens=1024, stop_sequences=None
 
 
 class FaraLocalModel(Model):
+    """
+    Wrapper for the local Fara (Qwen2.5-VL) model to work with SmolAgents.
+    """
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
 
     def __call__(
         self,
-        messages: List[Dict[str, Any]],
+        messages: List[ChatMessage],
         stop_sequences: Optional[List[str]] = None,
+        grammar: Optional[str] = None,
         **kwargs,
     ) -> ChatMessage:
         
         formatted_messages = []
+        
         for msg in messages:
-            role = msg["role"]
-            content = msg["content"]
+            # Safely access role and content from ChatMessage object using attributes
+            role = msg.role if hasattr(msg, "role") else "user"
+            content = msg.content if hasattr(msg, "content") else ""
+            
             new_content = []
             
             if isinstance(content, str):
@@ -170,6 +177,7 @@ class FaraLocalModel(Model):
                     elif isinstance(item, dict):
                         if "type" in item:
                             if item["type"] == "image":
+                                # Handle path or url - extract value to ensure serializability
                                 val = item.get("image") or item.get("url") or item.get("path")
                                 new_content.append({"type": "image", "image": val})
                             else: