Update app.py

app.py

@@ -4,7 +4,7 @@ from huggingface_hub import hf_hub_download
 import os
 
 # Hugging Face API token and repo setup
-hf_token = os.getenv("hfkeymeow")  # Make sure your token is correctly stored
+hf_token = os.getenv("hfkeymeow")  # Ensure your token is correctly stored
 repo_id = "conhllnd/Playper-V2"  # Repository ID on Hugging Face
 repo_type = "space"  # Repository type
 
@@ -57,17 +57,16 @@ def calculate_engagement_increase():
 
     return f"Leonardo got you {int(engagement_increase)} more engagements across TikTok & Facebook!"
 
-
 # Function to load and plot the masterpiece data (video and graph)
 def load_masterpiece_video_and_plot():
     # Download the video file from the Hugging Face repository
-    masterpiece_video_path = hf_hub_download(repo_id="conhllnd/Playper-V2", filename="starperformer_viral.mp4", repo_type="space")
+    masterpiece_video_path = download_file_from_repo("starperformer_viral.mp4")
 
     # Replace [actual outcome] with the correct label (here it's 'viral' from the filename)
     actual_outcome = "viral"
     
-    # Load the data for the TikTok graph (simulated example)
-    tiktok_csv_path = hf_hub_download(repo_id="conhllnd/Playper-V2", filename="tiktok_histogram.csv", repo_type="space")
+    # Load the data for the TikTok graph
+    tiktok_csv_path = download_file_from_repo("tiktok_histogram.csv")
     df = pd.read_csv(tiktok_csv_path)
 
     # Assuming the CSV has columns 'Engagement' and 'Density'
@@ -141,19 +140,19 @@ with gr.Blocks() as dashboard:
     # Section for TikTok projections (Videos)
     with gr.Row():
         with gr.Column():
-            video_path_1, graph_1 = load_data_and_plot("video", "TikTok1", tiktok_csv_path, video_folder)
+            video_path_1, graph_1 = load_data_and_plot("video", "TikTok1", download_file_from_repo("tiktok_histogram.csv"), download_file_from_repo("video_folder"))
             if video_path_1:
                 gr.Video(video_path_1, label="TikTok 1 Video")
                 gr.Plot(graph_1, label="TikTok 1 Performance")
 
         with gr.Column():
-            video_path_2, graph_2 = load_data_and_plot("video", "TikTok2", tiktok_csv_path, video_folder)
+            video_path_2, graph_2 = load_data_and_plot("video", "TikTok2", download_file_from_repo("tiktok_histogram.csv"), download_file_from_repo("video_folder"))
             if video_path_2:
                 gr.Video(video_path_2, label="TikTok 2 Video")
                 gr.Plot(graph_2, label="TikTok 2 Performance")
 
         with gr.Column():
-            video_path_3, graph_3 = load_data_and_plot("video", "TikTok3", tiktok_csv_path, video_folder)
+            video_path_3, graph_3 = load_data_and_plot("video", "TikTok3", download_file_from_repo("tiktok_histogram.csv"), download_file_from_repo("video_folder"))
             if video_path_3:
                 gr.Video(video_path_3, label="TikTok 3 Video")
                 gr.Plot(graph_3, label="TikTok 3 Performance")
@@ -161,19 +160,19 @@ with gr.Blocks() as dashboard:
     # Section for Facebook projections (Statics)
     with gr.Row():
         with gr.Column():
-            image_path_1, graph_4 = load_data_and_plot("image", "Static1", facebook_csv_path, image_folder)
+            image_path_1, graph_4 = load_data_and_plot("image", "Static1", download_file_from_repo("facebook_histogram.csv"), download_file_from_repo("image_folder"))
             if image_path_1:
                 gr.Image(image_path_1, label="Facebook Static 1")
                 gr.Plot(graph_4, label="Facebook Static 1 Performance")
 
         with gr.Column():
-            image_path_2, graph_5 = load_data_and_plot("image", "Static2", facebook_csv_path, image_folder)
+            image_path_2, graph_5 = load_data_and_plot("image", "Static2", download_file_from_repo("facebook_histogram.csv"), download_file_from_repo("image_folder"))
             if image_path_2:
                 gr.Image(image_path_2, label="Facebook Static 2")
                 gr.Plot(graph_5, label="Facebook Static 2 Performance")
 
         with gr.Column():
-            image_path_3, graph_6 = load_data_and_plot("image", "Static3", facebook_csv_path, image_folder)
+            image_path_3, graph_6 = load_data_and_plot("image", "Static3", download_file_from_repo("facebook_histogram.csv"), download_file_from_repo("image_folder"))
             if image_path_3:
                 gr.Image(image_path_3, label="Facebook Static 3")
                 gr.Plot(graph_6, label="Facebook Static 3 Performance")