Update app.py

app.py

@@ -3,7 +3,7 @@ import regex as re
 import csv
 import pandas as pd
 from analyzer import combine_repo_files_for_llm, analyze_combined_file, parse_llm_json_response
-from hf_utils import download_space_repo
+from hf_utils import download_space_repo, search_top_spaces
 
 # from hf_utils import download_space_repo
 
@@ -48,6 +48,22 @@ def process_repo_input_and_store(text):
     df = read_csv_as_text(csv_filename)
     return df
 
+def keyword_search_and_update(keyword):
+    global last_repo_ids, current_repo_idx
+    if not keyword:
+        return pd.DataFrame(columns=["repo id", "strength", "weaknesses", "speciality", "relevance rating"])
+    repo_ids = search_top_spaces(keyword, limit=5)
+    last_repo_ids = repo_ids
+    current_repo_idx = 0
+    csv_filename = "repo_ids.csv"
+    with open(csv_filename, mode="w", newline='', encoding="utf-8") as csvfile:
+        writer = csv.writer(csvfile)
+        writer.writerow(["repo id", "strength", "weaknesses", "speciality", "relevance rating"])
+        for repo_id in repo_ids:
+            writer.writerow([repo_id, "", "", "", ""])
+    df = read_csv_as_text(csv_filename)
+    return df
+
 def show_combined_repo_and_llm():
     global current_repo_idx
     if not last_repo_ids:
@@ -108,6 +124,12 @@ with gr.Blocks() as demo:
     submit_btn = gr.Button("Submit Repo IDs")
     submit_btn.click(process_repo_input_and_store, inputs=repo_id_box, outputs=df_box)
 
+    gr.Markdown("---")
+    gr.Markdown("## Keyword Search for Repos")
+    keyword_input = gr.Textbox(label="Enter keyword to search repos", lines=1, placeholder="e.g. audio")
+    keyword_btn = gr.Button("Search and Update Repo List")
+    keyword_btn.click(keyword_search_and_update, inputs=keyword_input, outputs=df_box)
+
     gr.Markdown("---")
     gr.Markdown("## Combine and Display Repo Files")
     combine_btn = gr.Button("Download, Combine & Show .py/.md Files from Next Repo and Analyze")