Update app.py

app.py

@@ -77,7 +77,7 @@ def analyze_with_gemini(posts):
             sentiments.append({
                 "Post": post,
                 "Sentiment": label,
-                "Confidence": 0.95  # Gemini usually doesn’t return prob, we keep 0.95
+                "Confidence": 0.95
             })
         except:
             sentiments.append({
@@ -92,10 +92,13 @@ def analyze_with_gemini(posts):
 # -----------------------------
 def run_analysis(hashtag, n_posts, vis_type, use_gemini):
     posts = generate_fake_posts(hashtag, n_posts)
+    
     if use_gemini:
         data = analyze_with_gemini(posts)
+        source_info = f"Analyzed with Gemini AI: {len(posts)} posts"
     else:
         data = analyze_with_hf(posts)
+        source_info = f"Analyzed with Hugging Face Transformers: {len(posts)} posts"
     
     df = pd.DataFrame(data)
 
@@ -109,7 +112,8 @@ def run_analysis(hashtag, n_posts, vis_type, use_gemini):
     else:
         fig = px.scatter(df, x="Sentiment", y="Confidence", title=f"Scatter of Sentiments for {hashtag}")
 
-    return df, fig
+    return df, fig, source_info
+
 
 # -----------------------------
 # Gradio UI
@@ -129,11 +133,12 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange", secondary_hue="blue"))
         with gr.Column(scale=2):
             output_table = gr.Dataframe(headers=["Post", "Sentiment", "Confidence"], wrap=True)
             output_plot = gr.Plot()
+            source_label = gr.Label(label="Analysis Source")
 
     run_btn.click(
         fn=run_analysis,
         inputs=[hashtag, n_posts, vis_type, use_gemini],
-        outputs=[output_table, output_plot]
+        outputs=[output_table, output_plot, source_label]
     )
 
 # -----------------------------