fix plot update button

.gitignore

@@ -1 +1,2 @@
-venv
+venv/
+__pycache__/

app.py

@@ -5,22 +5,37 @@ import gradio as gr
 
 def run_scripts():
     try:
-        # Execute both scripts
-        subprocess.run(
+        # Execute datasets script
+        result = subprocess.run(
             ["python", "hub_datasets_by_language.py"],
             capture_output=True,
             text=True,
             check=True,
         )
-        # subprocess.run(
+
+        # Uncomment this when models script is ready
+        # models_result = subprocess.run(
         #     ["python", "hub_models_by_language.py"],
         #     capture_output=True,
         #     text=True,
         #     check=True,
         # )
-        return "Scripts executed successfully! All plots have been updated."
+
+        # Return success message and updated image paths
+        return (
+            "✅ Scripts executed successfully! All plots have been updated.",
+            "plots/datasets_bar_plot_horizontal.png",
+            "plots/datasets_stack_area_en_es.png",
+            "plots/datasets_bar_plot_vertical.png",
+            "plots/datasets_time_series.png",
+        )
     except subprocess.CalledProcessError as e:
-        return f"Failed to execute scripts: {str(e.stderr)}"
+        error_msg = e.stderr if e.stderr else e.stdout if e.stdout else str(e)
+        # Return error message and keep current images
+        return (f"❌ Failed to execute scripts: {error_msg}", None, None, None, None)
+    except Exception as e:
+        # Return error message and keep current images
+        return (f"❌ Unexpected error: {str(e)}", None, None, None, None)
 
 
 def create_app():
@@ -43,14 +58,14 @@ def create_app():
 
         with gr.Row():
             with gr.Column():
-                gr.Image(
+                img1 = gr.Image(
                     value="plots/datasets_bar_plot_horizontal.png",
                     label="Distribution of Datasets by Year (Horizontal)",
                     show_label=True,
                     show_download_button=True,
                     show_share_button=True,
                 )
-                gr.Image(
+                img2 = gr.Image(
                     value="plots/datasets_stack_area_en_es.png",
                     label="Cumulative Growth of Datasets (Stacked)",
                     show_label=True,
@@ -58,14 +73,14 @@ def create_app():
                     show_share_button=True,
                 )
             with gr.Column():
-                gr.Image(
+                img3 = gr.Image(
                     value="plots/datasets_bar_plot_vertical.png",
                     label="Distribution of Datasets by Year (Vertical)",
                     show_label=True,
                     show_download_button=True,
                     show_share_button=True,
                 )
-                gr.Image(
+                img4 = gr.Image(
                     value="plots/datasets_time_series.png",
                     label="Cumulative Growth of Datasets (Line)",
                     show_label=True,
@@ -89,7 +104,7 @@ def create_app():
         #     )
 
         with gr.Row():
-            update_button = gr.Button("Update Plots with Latest Data")
+            update_button = gr.Button("Update plots with latest data (5 mins)")
             output_label = gr.Label()
 
         gr.Markdown(
@@ -120,7 +135,7 @@ def create_app():
 
         update_button.click(
             fn=run_scripts,
-            outputs=output_label,
+            outputs=[output_label, img1, img2, img3, img4],
         )
 
     return app

datasets_cache.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a139fa06dfe21c909136c004dc91e0dc0a92e81ffb7ca68fc6c1353d8717851c
-size 33831411
+oid sha256:b8573313007cdefe69256cd004c8a1b3d46b96dc2aed53cfe97fd9ae6235a6c9
+size 45749835