Update app.py

app.py

@@ -5,21 +5,28 @@ from src import search_on_web, search_custom_sites, search_custom_domain
 
 
 def run_search_on_web(query: str):
-    result = search_on_web(query)
-    return result.response_str, [item.model_dump() for item in result.citation]
-
+    try:
+        result = search_on_web(query)
+        return result.response_str, [item.model_dump() for item in result.citation]
+    except Exception as e:
+        return str(e), []
 
 def run_search_custom_sites(query: str, sites: str):
-    site_list = [s.strip() for s in sites.split(",") if s.strip()]
-    result = search_custom_sites(query, site_list)
-    return result.response_str, [item.model_dump() for item in result.citation]
+    try:
+        site_list = [s.strip() for s in sites.split(",") if s.strip()]
+        result = search_custom_sites(query, site_list)
+        return result.response_str, [item.model_dump() for item in result.citation]
+    except Exception as e:
+        return str(e), []
 
 
 def run_search_custom_domains(query: str, domains: str):
-    domain_list = [d.strip() for d in domains.split(",") if d.strip()]
-    result = search_custom_domain(query, domain_list)
-    return result.response_str, [item.model_dump() for item in result.citation]
-
+    try:
+        domain_list = [d.strip() for d in domains.split(",") if d.strip()]
+        result = search_custom_domain(query, domain_list)
+        return result.response_str, [item.model_dump() for item in result.citation]
+    except Exception as e:
+        return str(e), []
 
 with gr.Blocks() as demo:
     gr.Markdown("# 🔎 MCP Search Panel")