Update app.py

app.py

@@ -1,5 +1,21 @@
-import ast
+import gradio as gr
+from smolagents import HfApiModel, CodeAgent
+from huggingface_hub import login
+import os
+import shutil
+import ast  # Add missing import
 
+# Add this before model initialization
+hf_token = os.getenv("HF_TOKEN")
+login(token=hf_token, add_to_git_credential=True)
+
+# Then create model with explicit token
+model = HfApiModel(
+    "mistralai/Mixtral-8x7B-Instruct-v0.1",
+    token=hf_token  # Pass token explicitly
+)
+
+# Add this formatting function
 def format_analysis_report(raw_output, visuals):
     try:
         # Check if raw_output is already a dictionary
@@ -27,4 +43,12 @@ def format_analysis_report(raw_output, visuals):
         return report, visuals
     except Exception as e:
         print(f"Error formatting report: {e}")
-        return raw_output, visuals
+        return raw_output, visuals
+
+def format_observations(observations):
+    items = []
+    for key, value in observations.items():
+        if 'proportions' in key:
+            items.append(f"""
+            <div style="margin: 15px 0; padding: 15px; background: white; border-radius: 8px; box-shadow: 0 2px 4px rgba(0,0,0,0.05);">
+                <h3 style="margin: 0 0 10px 0; color: #4A708