Update app.py

app.py

@@ -193,23 +193,49 @@ def SwarmyTime(data: List[Dict]) -> Dict:
     aggregated["unique_authors"] = len(aggregated["unique_authors"])
     return aggregated
 
-with gr.Blocks() as demo:
-    gr.Markdown("## Search the Hugging Face Hub")
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("""
+    # Search the Hugging Face Hub
+    Search and download models, datasets, and spaces from Hugging Face.
+    """)
+    
     with gr.Row():
-        search_query = gr.Textbox(label="Search Query", value="awacke1")
-        search_type = gr.Radio(["Models", "Datasets", "Spaces"], label="Search Type", value="Models")
-        search_button = gr.Button("Search")
+        with gr.Column(scale=3):
+            search_query = gr.Textbox(
+                label="Search Query",
+                value="awacke1",
+                placeholder="Enter search term..."
+            )
+        with gr.Column(scale=2):
+            search_type = gr.Radio(
+                ["Models", "Datasets", "Spaces"],
+                label="Search Type",
+                value="Models",
+                container=True
+            )
+        with gr.Column(scale=1):
+            search_button = gr.Button("🔍 Search", variant="primary", scale=1)
     
-    results_html = gr.HTML(label="Search Results")
+    with gr.Row(variant="panel"):
+        with gr.Column(scale=1):
+            gr.Markdown("### Download Options")
+            with gr.Row():
+                download_readme_button = gr.Button(
+                    "📚 Download READMEs",
+                    variant="secondary",
+                )
+                download_repo_button = gr.Button(
+                    "📦 Download Repositories",
+                    variant="secondary",
+                )
+            download_status = gr.Markdown("Status: Ready to download", label="Status")
+            download_area = gr.HTML("", label="Download Link")
     
     with gr.Row():
-        download_readme_button = gr.Button("📚 Download READMEs", visible=False)
-        download_repo_button = gr.Button("📦 Download Repositories", visible=False)
-    
-    download_status = gr.Markdown("", label="Download Status")
-    download_area = gr.HTML("", label="Download Link")
-    metadata_output = gr.Textbox(label="Metadata", lines=10)
-    aggregated_output = gr.JSON(label="Aggregated Content")
+        with gr.Column(scale=2):
+            results_html = gr.HTML(label="Search Results")
+        with gr.Column(scale=1):
+            aggregated_output = gr.JSON(label="Search Statistics")
     
     search_type_state = gr.State("")
     current_results = gr.State([])
@@ -219,12 +245,9 @@ with gr.Blocks() as demo:
         data = df.to_dict('records')
         aggregated = SwarmyTime(data)
         html_results = display_results(df)
-        show_download = len(data) > 0
         return [
             html_results,  # results_html
-            show_download,  # download_readme_button visible
-            show_download,  # download_repo_button visible
-            "",  # download_status
+            "Status: Ready to download",  # download_status
             "",  # download_area
             aggregated,  # aggregated_output
             search_type,  # search_type_state
@@ -233,23 +256,21 @@ with gr.Blocks() as demo:
 
     async def handle_readme_download(data):
         if not data:
-            return ["No results to download", ""]
+            return ["Status: No results to download", ""]
         download_link, status = await download_all_readmes(data)
-        return [status, download_link]
+        return [f"Status: {status}", download_link]
 
     def handle_repo_download(data, search_type):
         if not data:
-            return ["No results to download", ""]
+            return ["Status: No results to download", ""]
         download_link, status = create_repo_zip(data, search_type)
-        return [status, download_link]
+        return [f"Status: {status}", download_link]
 
     search_button.click(
         search_and_aggregate,
         inputs=[search_query, search_type],
         outputs=[
             results_html,
-            download_readme_button,
-            download_repo_button,
             download_status,
             download_area,
             aggregated_output,