connect script to app and update plots

app.py

@@ -1,27 +1,28 @@
+import subprocess
+
 import gradio as gr
-import papermill as pm
 
 
-def run_notebook():
+def run_script():
     try:
-        # Execute the notebook
-        pm.execute_notebook(
-            "hub_datasets_by_language.ipynb",
-            "hub_datasets_by_language_output.ipynb",  # Save the output in a new notebook
+        result = subprocess.run(
+            ["python", "hub_datasets_by_language.py"],
+            capture_output=True,
+            text=True,
+            check=True,
         )
-        return "Notebook executed successfully!"
-    except Exception as e:
-        return f"Failed to execute notebook: {str(e)}"
+        return "Script executed successfully! Plots have been updated."
+    except subprocess.CalledProcessError as e:
+        return f"Failed to execute script: {str(e.stderr)}"
 
 
 def create_app():
     with gr.Blocks() as app:
-
         gr.Markdown(
             """
             # Visualizing The Language Gap In The Hugging Face Hub
                     
-            The open-source community is creating more a more resources in languages other than English but there is still a huge gap. This Space showcases plots that can help visualize this gap in the case of Spanish and can easily be adapted to other languages.
+            The open-source community is creating more and more resources in languages other than English but there is still a huge gap. This Space showcases plots that can help visualize this gap in the case of Spanish and can easily be adapted to other languages.
         """
         )
 
@@ -32,49 +33,50 @@ def create_app():
             Note: We consider only **monolingual** datasets in these plots, i.e. datasets that only contain data in one language. This is because *most* of the multilingual datasets are usually machine-translated and we want to focus on original data.        
         """
         )
+
         with gr.Row():
             with gr.Column():
-                image1 = gr.Image(
+                gr.Image(
                     value="plots/bar_plot_horizontal.png",
-                    label="Bar Plot Horizontal",
+                    label="Distribution by Year (Horizontal)",
                     show_label=True,
                     show_download_button=True,
                     show_share_button=True,
                 )
-                image2 = gr.Image(
-                    value="plots/bar_plot_vertical.png",
-                    label="Bar Plot Vertical",
+                gr.Image(
+                    value="plots/stack_area_en_es.png",
+                    label="Stacked Area Plot",
                     show_label=True,
                     show_download_button=True,
                     show_share_button=True,
                 )
             with gr.Column():
-                image3 = gr.Image(
-                    value="plots/stack_area.png",
-                    label="Stack Area",
+                gr.Image(
+                    value="plots/bar_plot_vertical.png",
+                    label="Distribution by Year (Vertical)",
                     show_label=True,
                     show_download_button=True,
                     show_share_button=True,
                 )
-                image4 = gr.Image(
+                gr.Image(
                     value="plots/time_series.png",
-                    label="Time Series",
+                    label="Cumulative Growth Over Time",
                     show_label=True,
                     show_download_button=True,
                     show_share_button=True,
                 )
 
+        with gr.Row():
+            update_button = gr.Button("Update Plots with Latest Data")
+            output_label = gr.Label()
+
         gr.Markdown(
             """
             ## Adapt to other languages
 
-            This Space is WIP and more languages and visuals will be included shortly. Meanwhile, you can clone the Space, adapt the code in the notebook and run it to generate plots for other languages.
+            This Space is WIP and more languages and visuals will be included shortly. Meanwhile, you can clone the Space, adapt the code in the script and run it to generate plots for other languages.
             """
         )
-        run_button = gr.Button("Run Notebook")
-        output_label = gr.Label()  # Display the result of running the notebook
-
-        run_button.click(run_notebook, outputs=output_label)
 
         gr.Markdown("## Citation")
         with gr.Accordion("Citation information", open=False):
@@ -94,6 +96,11 @@ def create_app():
                 """
             )
 
+        update_button.click(
+            fn=run_script,
+            outputs=output_label,
+        )
+
     return app