Update app.py

app.py

@@ -13,7 +13,6 @@ from sklearn.ensemble import RandomForestClassifier
 from sklearn.model_selection import train_test_split, cross_val_score
 from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score
 from sklearn.preprocessing import LabelEncoder
-from smolagents.types import AgentText
 
 # Authenticate with Hugging Face
 hf_token = os.getenv("HF_TOKEN")
@@ -86,21 +85,17 @@ def run_agent(_):
         additional_args={"source_file": temp_file.name}
     )
 
-    # Convert AgentText object to string and parse it
-    if isinstance(result, AgentText):
-        import json
-        result_str = result.text.strip()
-
-        try:
-            result_dict = json.loads(result_str)
-        except json.JSONDecodeError:
-            return f"Error decoding agent response: {result_str}", []
+    # If the result is text, attempt to parse it as JSON
+    try:
+        result_str = result.strip()
+        result_dict = json.loads(result_str)
 
         insights = result_dict.get("insights", "No insights generated.")
         image_paths = result_dict.get("figures", [])
         return insights, image_paths
-    else:
-        return "Unexpected result type from agent", []
+    except Exception as e:
+        return f"Error parsing agent response: {e}", []
+