Update app.py

app.py

@@ -2,8 +2,7 @@ import gradio as gr
 import pandas as pd
 from huggingface_hub import hf_hub_download
 import os
-import plotly.graph_objects as go  # Import Plotly for graph plotting
-
+import plotly.graph_objects as go
 
 # Hugging Face API token and repo setup
 hf_token = os.getenv("hfkeymeow")  # Ensure your token is correctly stored
@@ -106,6 +105,28 @@ def highlight_masterpiece_quintile(fig, actual_outcome, x_max):
     )
     return fig
 
+# Function to load and plot data (TikTok, Facebook)
+def load_data_and_plot(media_type, media_identifier, csv_path, folder_path):
+    df = pd.read_csv(csv_path)
+
+    # Assuming the CSV has columns 'Engagement' and 'Density'
+    x_data = df['Engagement']
+    y_data = df['Density']
+
+    # Create the plot with actual data
+    fig = go.Figure()
+    fig.add_trace(go.Scatter(x=x_data, y=y_data, mode='lines', name="Actual Data"))
+
+    # Adjust the x-axis and y-axis
+    fig.update_layout(
+        title=f"Predicted Engagement for {media_identifier}",
+        xaxis=dict(range=[0, max(x_data)]),
+        yaxis=dict(range=[0, max(y_data)])
+    )
+
+    media_file_path = None  # Assuming you aren't showing media in this function
+    return media_file_path, fig
+
 # Create the Gradio interface
 with gr.Blocks() as dashboard:
 
@@ -142,7 +163,7 @@ 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", download_file_from_repo("tiktok_histogram.csv"), download_file_from_repo("video_folder"))
+            video_path_1, graph_1 = load_data_and_plot("video", "TikTok1", download_file_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")
@@ -181,3 +202,4 @@ with gr.Blocks() as dashboard:
 
 # Launch the dashboard
 dashboard.launch(debug=True)
+