Upload app.py

app.py

@@ -78,48 +78,85 @@ def advanced_search_company(company_input: str) -> dict:
     result = financial_analyzer.search_company(company_input)
     return result if not result.get("error") else {"error": result["error"]}
 
-# Gradio wrapper functions
+# Gradio wrapper functions (添加调试和超时处理)
 def gradio_search_company(company_name: str):
     """Gradio wrapper for search_company"""
     if not company_name or not company_name.strip():
         return json.dumps({"error": "Company name cannot be empty"}, indent=2)
     try:
+        import sys
+        print(f"[DEBUG] Searching company: {company_name.strip()}", file=sys.stderr)
         result = search_company(company_name.strip())
+        print(f"[DEBUG] Search result type: {type(result)}", file=sys.stderr)
+        print(f"[DEBUG] Search result: {result}", file=sys.stderr)
+        # Ensure result is a dict
+        if not isinstance(result, dict):
+            result = {"error": f"Unexpected result type: {type(result)}"}
         return json.dumps(result, indent=2)
+    except TimeoutError as e:
+        return json.dumps({"error": f"Request timeout: {str(e)}"}, indent=2)
     except Exception as e:
-        return json.dumps({"error": f"Exception: {str(e)}"}, indent=2)
+        import traceback
+        traceback.print_exc()
+        return json.dumps({"error": f"Exception: {str(e)}", "type": str(type(e))}, indent=2)
 
 def gradio_get_company_info(cik: str):
     """Gradio wrapper for get_company_info"""
     if not cik or not cik.strip():
         return json.dumps({"error": "CIK cannot be empty"}, indent=2)
     try:
+        import sys
+        print(f"[DEBUG] Getting company info for CIK: {cik.strip()}", file=sys.stderr)
         result = get_company_info(cik.strip())
+        print(f"[DEBUG] Company info result: {result}", file=sys.stderr)
+        if not isinstance(result, dict):
+            result = {"error": f"Unexpected result type: {type(result)}"}
         return json.dumps(result, indent=2)
+    except TimeoutError as e:
+        return json.dumps({"error": f"Request timeout: {str(e)}"}, indent=2)
     except Exception as e:
-        return json.dumps({"error": f"Exception: {str(e)}"}, indent=2)
+        import traceback
+        traceback.print_exc()
+        return json.dumps({"error": f"Exception: {str(e)}", "type": str(type(e))}, indent=2)
 
 def gradio_extract_metrics(cik: str, years: float):
     """Gradio wrapper for extract_financial_metrics"""
     if not cik or not cik.strip():
         return json.dumps({"error": "CIK cannot be empty"}, indent=2)
     try:
-        # Convert float slider value to int
+        import sys
         years_int = int(years)
+        print(f"[DEBUG] Extracting metrics for CIK: {cik.strip()}, Years: {years_int}", file=sys.stderr)
         result = extract_financial_metrics(cik.strip(), years_int)
+        print(f"[DEBUG] Extract metrics result: {result}", file=sys.stderr)
+        if not isinstance(result, dict):
+            result = {"error": f"Unexpected result type: {type(result)}"}
         return json.dumps(result, indent=2)
+    except TimeoutError as e:
+        return json.dumps({"error": f"Request timeout: {str(e)}"}, indent=2)
     except Exception as e:
-        return json.dumps({"error": f"Exception: {str(e)}"}, indent=2)
+        import traceback
+        traceback.print_exc()
+        return json.dumps({"error": f"Exception: {str(e)}", "type": str(type(e))}, indent=2)
 
 def gradio_get_latest(cik: str):
     """Gradio wrapper for get_latest_financial_data"""
     if not cik or not cik.strip():
         return json.dumps({"error": "CIK cannot be empty"}, indent=2)
     try:
+        import sys
+        print(f"[DEBUG] Getting latest data for CIK: {cik.strip()}", file=sys.stderr)
         result = get_latest_financial_data(cik.strip())
+        print(f"[DEBUG] Latest data result: {result}", file=sys.stderr)
+        if not isinstance(result, dict):
+            result = {"error": f"Unexpected result type: {type(result)}"}
         return json.dumps(result, indent=2)
+    except TimeoutError as e:
+        return json.dumps({"error": f"Request timeout: {str(e)}"}, indent=2)
     except Exception as e:
-        return json.dumps({"error": f"Exception: {str(e)}"}, indent=2)
+        import traceback
+        traceback.print_exc()
+        return json.dumps({"error": f"Exception: {str(e)}", "type": str(type(e))}, indent=2)
 
 # Create Gradio interface
 with gr.Blocks(title="SEC Financial Data MCP Server", theme=gr.themes.Soft()) as demo: