Update app.py

app.py

@@ -64,15 +64,8 @@ def guarded_fetch_analytics():
     return count_md, plot
 
 def run_mentions_and_load():
-    try:
-        html_path = generate_mentions_dashboard(
-        token_received["client_id"],
-        token_received["token"]
-    )
-        with open(html_path, "r", encoding="utf-8") as f:
-            return f.read()
-    except Exception as e:
-        return f"<p style='color:red;'>Failed to load mentions: {e}</p>"
+    html, fig = generate_mentions_dashboard(comm_client_id, comm_token_dict)
+    return html, fig
 
 
 # --- Build the Gradio UI ---
@@ -131,14 +124,16 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
         with gr.TabItem("3️⃣ Mentions"):
             gr.Markdown("Analyze sentiment of recent posts that mention your organization.")
             fetch_mentions_btn = gr.Button("🧠 Fetch Mentions & Sentiment", variant="primary")
-            mentions_html = gr.HTML(value="<p style='text-align:center; color:#555;'>Waiting for token...</p>")
+            mentions_html = gr.HTML()
+            mentions_plot = gr.Plot(visible=True)
             fetch_mentions_btn.click(
                 fn=run_mentions_and_load,
                 inputs=[],
-                outputs=[mentions_html]
+                outputs=[mentions_html, mentions_plot]
             )
 
 
+
 # Launch the app
 if __name__ == "__main__":
     app.launch(server_name="0.0.0.0", server_port=7860, share=True)