Update app.py

app.py

@@ -1,14 +1,14 @@
-from sentiment_analysis import SentimentAnalysisTool
+from simple_sentiment import SimpleSentimentTool
 
-# Create an instance of the tool
-sentiment_tool = SentimentAnalysisTool()
+# Create an instance of the tool without preloading to avoid startup errors
+sentiment_tool = SimpleSentimentTool(default_model="distilbert", preload=False)
 
 # Launch the Gradio interface
 if __name__ == "__main__":
     import gradio as gr
     
     with gr.Blocks(title="Sentiment Analysis Tool") as demo:
-        gr.Markdown("# Sentiment Analysis Tool")
+        gr.Markdown("# Multi-Model Sentiment Analysis Tool")
         
         with gr.Row():
             with gr.Column():
@@ -18,6 +18,12 @@ if __name__ == "__main__":
                     lines=5
                 )
                 
+                model_dropdown = gr.Dropdown(
+                    choices=list(sentiment_tool.models.keys()),
+                    value=sentiment_tool.default_model,
+                    label="Select Model"
+                )
+                
                 with gr.Row():
                     analyze_btn = gr.Button("Analyze Sentiment")
                     clear_btn = gr.Button("Clear")
@@ -25,9 +31,15 @@ if __name__ == "__main__":
             with gr.Column():
                 output = gr.JSON(label="Sentiment Analysis Results")
         
+        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)
+        
         analyze_btn.click(
-            fn=sentiment_tool,
-            inputs=text_input,
+            fn=analyze_with_model,
+            inputs=[text_input, model_dropdown],
             outputs=output
         )
         
@@ -39,11 +51,13 @@ if __name__ == "__main__":
         
         gr.Examples(
             examples=[
-                ["I love this product! It's amazing and works perfectly."],
-                ["This movie was terrible. I was very disappointed."],
-                ["The service was okay, but could be improved in several ways."]
+                ["I love this product! It's amazing and works perfectly.", "distilbert"],
+                ["This movie was terrible. I was very disappointed.", "distilbert"],
+                ["The service was okay, but could be improved in several ways.", "distilbert"],
+                ["Ce produit est vraiment excellent!", "multilingual"],
+                ["Dieses Buch ist sehr interessant.", "german"]
             ],
-            inputs=text_input
+            inputs=[text_input, model_dropdown]
         )
     
     demo.launch(share=True)