add total

app.py

@@ -21,6 +21,10 @@ def get_model_stats(search_term):
         sort="_id"  # Sort by ID to avoid timeout issues
     )
     
+    # Initialize counters for total downloads
+    total_30day_downloads = 0
+    total_alltime_downloads = 0
+    
     # Create and write to CSV
     with open(output_file, 'w', newline='', encoding='utf-8') as csvfile:
         csv_writer = csv.writer(csvfile)
@@ -30,24 +34,39 @@ def get_model_stats(search_term):
         # Process models
         model_count = 0
         for model in models_generator:
+            # Get download counts
+            downloads_30day = getattr(model, 'downloads', 0)
+            downloads_alltime = getattr(model, 'downloads_all_time', 0)
+            
+            # Add to totals
+            total_30day_downloads += downloads_30day
+            total_alltime_downloads += downloads_alltime
+            
             # Write to CSV
             csv_writer.writerow([
                 getattr(model, 'id', "Unknown"),
-                getattr(model, 'downloads', 0),  # Last 30 days downloads
-                getattr(model, 'downloads_all_time', 0)
+                downloads_30day,
+                downloads_alltime
             ])
             model_count += 1
     
     # Read the CSV file into a pandas DataFrame
     df = pd.read_csv(output_file)
     
+    # Create status message with total downloads
+    status_message = (
+        f"Found {model_count} models for search term '{search_term}'\n"
+        f"Total 30-day downloads: {total_30day_downloads:,}\n"
+        f"Total all-time downloads: {total_alltime_downloads:,}"
+    )
+    
     # Return both the DataFrame, status message, and the CSV file path
-    return df, f"Found {model_count} models for search term '{search_term}'", str(output_file)
+    return df, status_message, str(output_file)
 
 # Create the Gradio interface
 with gr.Blocks(title="Hugging Face Model Statistics") as demo:
     gr.Markdown("# Hugging Face Model Statistics")
-    gr.Markdown("Enter a search term to find model statistics from the Hub")
+    gr.Markdown("Enter a search term to find model statistics from Hugging Face Hub")
     
     with gr.Row():
         search_input = gr.Textbox(
@@ -63,7 +82,7 @@ with gr.Blocks(title="Hugging Face Model Statistics") as demo:
             datatype=["str", "number", "number"],
             label="Model Statistics"
         )
-        status_message = gr.Textbox(label="Status")
+        status_message = gr.Textbox(label="Status", lines=3)  # Increased lines to show all stats
     
     with gr.Row():
         download_button = gr.Button("Download CSV")