Update app.py

app.py

@@ -14,16 +14,6 @@ import tempfile
 import shutil
 import os
 
-# Search Terms
-example_search_terms = [
-    {"id": "gpt-3", "emoji": "🤖"},
-    {"id": "stable-diffusion", "emoji": "🎨"},
-    {"id": "whisper", "emoji": "🗣️"},
-    {"id": "bert", "emoji": "📖"},
-    {"id": "resnet", "emoji": "🖼️"}
-]
-
-
 # Initialize HuggingFace with access token
 def init_huggingface(token: str):
     """Initialize HuggingFace with access token."""
@@ -88,24 +78,6 @@ def SwarmyTime(data: List[Dict]) -> Dict:
     aggregated["unique_authors"] = len(aggregated["unique_authors"])
     return aggregated
 
-def search_and_aggregate(query, search_type, token, example_term):
-    if example_term:
-        query = example_term.split(" ")[1]  # Extract the user ID from the button label
-    df = search_hub(query, search_type, token)
-    data = df.to_dict('records')
-    aggregated = SwarmyTime(data)
-    html_results = display_results(df)
-    return [
-        html_results,
-        "Status: Ready to download",
-        "",
-        aggregated,
-        search_type,
-        data
-    ]
-
-
-
 def search_hub(query: str, search_type: str, token: str = None) -> pd.DataFrame:
     """Search the Hugging Face Hub for models, datasets, or spaces."""
     api = HfApi(token=token)
@@ -279,13 +251,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     Search and download models, datasets, and spaces from Hugging Face.
     """)
     
-    with gr.Row():
-        example_search_buttons = gr.Dataset(
-            components=[gr.Button(value=f"{term['emoji']} {term['id']}", variant="secondary") for term in example_search_terms],
-            label="Example Search Terms",
-            samples=[]
-        )
-    
     with gr.Row():
         with gr.Column(scale=3):
             hf_token = gr.Textbox(
@@ -348,49 +313,22 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             search_type,
             data
         ]
+
     async def handle_readme_download(data, token):
-        if data:
-            download_link, status = await download_all_readmes(data, token)
-            return [f"Status: {status}", download_link]
-        else:
+        if not data:
             return ["Status: No results to download", ""]
-    
+        download_link, status = await download_all_readmes(data, token)
+        return [f"Status: {status}", download_link]
+
     def handle_repo_download(data, search_type, token):
-        if data:
-            download_link, status = create_repo_zip(data, search_type, token)
-            return [f"Status: {status}", download_link]
-        else:
+        if not data:
             return ["Status: No results to download", ""]
-
-    #async def handle_readme_download(data, token):
-    #    if not data:
-    #        return ["Status: No results to download", ""]
-    #    download_link, status = await download_all_readmes(data, token)
-    #    return [f"Status: {status}", download_link]
-
-    #def handle_repo_download(data, search_type, token):
-    #    if not data:
-    #        return ["Status: No results to download", ""]
-    #    download_link, status = create_repo_zip(data, search_type, token)
-    #    return [f"Status: {status}", download_link]
-
-    #search_button.click(
-    #    search_and_aggregate,
-    #    inputs=[search_query, search_type, hf_token],
-    #    outputs=[
-    #        results_html,
-    #        download_status,
-    #        download_area,
-    #        aggregated_output,
-    #        search_type_state,
-    #        current_results
-    #    ]
-    #)
-
+        download_link, status = create_repo_zip(data, search_type, token)
+        return [f"Status: {status}", download_link]
 
     search_button.click(
         search_and_aggregate,
-        inputs=[search_query, search_type, hf_token, example_search_buttons],
+        inputs=[search_query, search_type, hf_token],
         outputs=[
             results_html,
             download_status,
@@ -398,10 +336,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             aggregated_output,
             search_type_state,
             current_results
-        ],
-        call_after=lambda data: [handle_readme_download(data[-1], hf_token.value), handle_repo_download(data[-1], search_type.value, hf_token.value)]
+        ]
     )
-    
+
     download_readme_button.click(
         handle_readme_download,
         inputs=[current_results, hf_token],