Update app.py

app.py

@@ -1,4 +1,4 @@
-from sentiment_analysis import SimpleSentimentTool
+from simple_sentiment import SimpleSentimentTool
 
 # Create an instance of the tool without preloading to avoid startup errors
 sentiment_tool = SimpleSentimentTool(default_model="distilbert", preload=False)
@@ -34,8 +34,15 @@ if __name__ == "__main__":
         def analyze_with_model(text, model_key):
             """Call the tool's forward method directly with appropriate parameters."""
             if not text:
-                return {"error": "Please enter some text to analyze"}
-            return sentiment_tool.forward(text, model_key)
+                return "{\"error\": \"Please enter some text to analyze\"}"
+            # The tool returns a JSON string now
+            json_str = sentiment_tool.forward(text, model_key)
+            # But we need to parse it for the Gradio JSON component
+            import json
+            try:
+                return json.loads(json_str)
+            except:
+                return {"error": "Failed to parse results"}
         
         analyze_btn.click(
             fn=analyze_with_model,