csv

app.py

@@ -4,6 +4,7 @@ import pandas as pd
 import gradio as gr
 from huggingface_hub import HfApi
 from pathlib import Path
+import os
 
 def get_model_stats(search_term):
     # Initialize the API
@@ -87,7 +88,7 @@ with gr.Blocks(title="Hugging Face Model Statistics") as demo:
     
     with gr.Row():
         download_button = gr.Button("Download CSV")
-        csv_file = gr.File(label="CSV File", visible=False)
+        csv_file = gr.File(label="CSV File", visible=True)
     
     # Store the CSV file path in a state
     csv_path = gr.State()
@@ -99,6 +100,11 @@ with gr.Blocks(title="Hugging Face Model Statistics") as demo:
         )
         return df, status, csv_path
     
+    def get_csv_file(csv_path):
+        if csv_path and os.path.exists(csv_path):
+            return csv_path
+        return None
+    
     search_button.click(
         fn=get_model_stats,
         inputs=search_input,
@@ -110,7 +116,7 @@ with gr.Blocks(title="Hugging Face Model Statistics") as demo:
     )
     
     download_button.click(
-        fn=lambda x: x,
+        fn=get_csv_file,
         inputs=csv_path,
         outputs=csv_file
     )