Update app.py

app.py

@@ -27,33 +27,33 @@ def documentation():
     Choose a model from the dropdown and enter text to see the sentiment prediction.
     """
 
-# Define example inputs
-exams = [
-    "I absolutely love this product! It has changed my life.",
-    "This is the worst movie I have ever seen. Completely disappointing.",
-    "I'm not sure how I feel about this new update. It has some good points, but also many drawbacks.",
-    "The customer service was fantastic! Very helpful and polite.",
-    "Honestly, this was quite a mediocre experience. Nothing special."
-]
-
-# Setup the tabs for Demo and Documentation
-tab1 = gr.Interface(
-    fn=predict_sentiment,
-    title="Sentiment Analysis",
-    description="Select a model and enter text to analyze sentiment.",
-    inputs=[gr.Textbox(label="Input Text", placeholder="Type here or click an example..."), 
-            gr.Radio(["Model 1 (RoBERTa-large)", "Model 2 (BERTweet)"], label="Model Choice")],
-    outputs="text",
-    examples=exams
-)
-
-# Setup the tabs for Documentation
-tab2 = gr.Interface(
-    fn=documentation,
-    title="Documentation",
-    inputs=None,
-    outputs=gr.Markdown()
-)
-
-iface = gr.TabbedInterface([tab1, tab2], ["Demo", "Documentation"])
-iface.launch()
+with gr.Blocks(title="Sentiment Analysis", theme=gr.themes.Soft()) as demo:
+    with gr.Tabs():
+        with gr.TabItem("Demo"):
+            with gr.Row():
+                with gr.Column(scale=2):
+                    text_input = gr.Textbox(label="Input Text", placeholder="Type here or select an example...")
+                    model_choice = gr.Radio(["Model 1 (RoBERTa-large)", "Model 2 (BERTweet)"], label="Model Choice", value="Model 1 (RoBERTa-large)")
+                    submit_button = gr.Button("Analyze")
+                with gr.Column():
+                    output = gr.Label()
+
+            examples = gr.Examples(examples=[
+                "I absolutely love this product! It has changed my life.",
+                "This is the worst movie I have ever seen. Completely disappointing.",
+                "I'm not sure how I feel about this new update. It has some good points, but also many drawbacks.",
+                "The customer service was fantastic! Very helpful and polite.",
+                "Honestly, this was quite a mediocre experience. Nothing special."
+            ], inputs=text_input)
+
+            submit_button.click(
+                predict_sentiment, 
+                inputs=[text_input, model_choice],
+                outputs=output
+            )
+        
+        with gr.TabItem("Documentation"):
+            doc_text = gr.Markdown()
+            doc_text.update(documentation())
+
+demo.launch()