Update app.py

app.py

@@ -3,39 +3,46 @@ import gradio as gr
 
 HF_API_BASE = "https://huggingface.co/api/models"
 SEARCH_TERMS = [
-    "MJ", "MichaelJackson", "Michael_Jackson", "Michael-Jackson", "michaeljackson",
-    "michael_jackson", "MJackson", "M_Jackson", "m-jackson", "mjackson",
-    "KingOfPop", "King_Of_Pop", "King-Of-Pop", "kingofpop"
+    "mj", "michaeljackson", "michael_jackson", "michael-jackson", "mjackson",
+    "m_jackson", "m-jackson", "kingofpop", "king_of_pop", "king-of-pop"
 ]
 
 def get_models(user: str):
     url = f"{HF_API_BASE}?search={user}"
-    response = requests.get(url)
-    if response.status_code == 200:
-        return response.json()
-    else:
+    headers = {"User-Agent": "Mozilla/5.0"}
+    try:
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()
+        return response.json() if response.text else []
+    except requests.exceptions.RequestException as e:
+        print(f"Error fetching models: {e}")
         return []
 
 def check_zip_files(model_id: str):
-    url = f"https://huggingface.co/{model_id}/tree/main"
-    response = requests.get(url)
+    url = f"https://huggingface.co/api/models/{model_id}/tree/main"
+    headers = {"User-Agent": "Mozilla/5.0"}
     zip_links = []
-    
-    if response.status_code == 200:
-        files = response.json()
-        for file in files:
-            if file['type'] == 'file' and file['path'].endswith('.zip'):
-                for term in SEARCH_TERMS:
-                    if term.lower() in file['path'].lower():
+    try:
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()
+        if response.text:
+            files = response.json()
+            for file in files:
+                file_path = file.get('path', '').lower()
+                if file.get('type') == 'file' and file_path.endswith('.zip'):
+                    if any(term in file_path or term in model_id.lower() for term in SEARCH_TERMS):
                         zip_links.append(f"https://huggingface.co/{model_id}/resolve/main/{file['path']}?download=true")
+    except requests.exceptions.RequestException as e:
+        print(f"Error accessing model {model_id}: {e}")
     return zip_links
 
 def search_huggingface_zips(user: str):
     models = get_models(user)
     all_links = []
     for model in models:
-        model_id = model['modelId']
-        all_links.extend(check_zip_files(model_id))
+        model_id = model.get('modelId', '')
+        if model_id and any(term in model_id.lower() for term in SEARCH_TERMS):
+            all_links.extend(check_zip_files(model_id))
     return "\n".join(all_links) if all_links else "No ZIP files found."
 
 demo = gr.Interface(
@@ -46,4 +53,4 @@ demo = gr.Interface(
     description="Enter a Hugging Face username to search for ZIP files related to Michael Jackson.",
 )
 
-demo.launch()
+demo.launch()