Update app.py

app.py

@@ -9,7 +9,7 @@ from mentions_dashboard import generate_mentions_dashboard
 from gradio_utils import get_url_user_token
 from Bubble_API_Calls import fetch_linkedin_token_from_bubble
 
-from linkedin_data_api_calls import fetch_linkedin_posts, fetch_linkedin_comments, analyze_sentiment, prepare_data_for_bubble, bulk_upload_to_bubble
+from Linkedin_Data_API_Calls import fetch_linkedin_posts, fetch_linkedin_comments, analyze_sentiment, prepare_data_for_bubble, bulk_upload_to_bubble
 
 # Fetch data
 posts = fetch_linkedin_posts(client_id, token_dict)
@@ -55,6 +55,24 @@ def process_and_store_bubble_token(url_user_token, org_urn, token_state):
 
     return check_token_status(new_state), new_state
 
+def guarded_fetch_posts(token_state):
+    if not token_state or not token_state.get("token"):
+        return "<p style='color:red; text-align:center;'>❌ Access denied. No token available.</p>"
+
+    client_id = token_state.get("client_id")
+    token_dict = token_state.get("token")
+
+    posts = fetch_linkedin_posts(client_id, token_dict)
+    comments_data = fetch_linkedin_comments(client_id, token_dict, [post["id"] for post in posts])
+    sentiments = analyze_sentiment(comments_data)
+    li_posts, li_post_stats, li_post_comments = prepare_data_for_bubble(posts, sentiments)
+
+    bulk_upload_to_bubble(li_posts, "LI_post")
+    bulk_upload_to_bubble(li_post_stats, "LI_post_stats")
+    bulk_upload_to_bubble(li_post_comments, "LI_post_comments")
+
+    return "<p style='color:green; text-align:center;'>✅ Posts and comments uploaded to Bubble.</p>"
+
 
 def guarded_fetch_dashboard(token_state):
     if not token_state or not token_state.get("token"):
@@ -96,6 +114,9 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
         outputs=[status_box, token_state]
     )
 
+    fetch_dashboard_btn.click(fn=guarded_fetch_posts, inputs=[token_state], outputs=[dashboard_html])
+
+    
     app.load(fn=check_token_status, inputs=[token_state], outputs=status_box)
     gr.Timer(5.0).tick(fn=check_token_status, inputs=[token_state], outputs=status_box)