Upload 2 files

app.py

@@ -16,7 +16,15 @@ def query_financial_data(company_name, query_type):
             json={"company_input": company_name},
             timeout=30
         )
-        company = search_resp.json()
+        
+        # 检查响应状态
+        if search_resp.status_code != 200:
+            return f"❌ Server Error: HTTP {search_resp.status_code}\n\nResponse: {search_resp.text[:500]}"
+        
+        try:
+            company = search_resp.json()
+        except ValueError as e:
+            return f"❌ JSON Parse Error: {str(e)}\n\nResponse Text: {search_resp.text[:500]}"
         
         if company.get("error"):
             return f"❌ 错误: {company['error']}"
@@ -34,7 +42,14 @@ def query_financial_data(company_name, query_type):
                 json={"cik": cik},
                 timeout=30
             )
-            data = data_resp.json()
+            
+            if data_resp.status_code != 200:
+                return result + f"❌ Server Error: HTTP {data_resp.status_code}\n\n{data_resp.text[:500]}"
+            
+            try:
+                data = data_resp.json()
+            except ValueError as e:
+                return result + f"❌ JSON Parse Error: {str(e)}\n\n{data_resp.text[:500]}"
             
             if data.get("error"):
                 return result + f"❌ {data['error']}"
@@ -51,7 +66,14 @@ def query_financial_data(company_name, query_type):
                 json={"cik": cik, "years": 3},
                 timeout=60
             )
-            metrics = metrics_resp.json()
+            
+            if metrics_resp.status_code != 200:
+                return result + f"❌ Server Error: HTTP {metrics_resp.status_code}\n\n{metrics_resp.text[:500]}"
+            
+            try:
+                metrics = metrics_resp.json()
+            except ValueError as e:
+                return result + f"❌ JSON Parse Error: {str(e)}\n\n{metrics_resp.text[:500]}"
             
             if metrics.get("error"):
                 return result + f"❌ {metrics['error']}"
@@ -77,7 +99,14 @@ def query_financial_data(company_name, query_type):
                 json={"cik": cik, "years": 5},
                 timeout=60
             )
-            metrics = metrics_resp.json()
+            
+            if metrics_resp.status_code != 200:
+                return result + f"❌ Server Error: HTTP {metrics_resp.status_code}\n\n{metrics_resp.text[:500]}"
+            
+            try:
+                metrics = metrics_resp.json()
+            except ValueError as e:
+                return result + f"❌ JSON Parse Error: {str(e)}\n\n{metrics_resp.text[:500]}"
             
             if metrics.get("error"):
                 return result + f"❌ {metrics['error']}"