Update app.py

app.py

@@ -140,16 +140,13 @@ Text:
         response = requests.post(
             modal_endpoint,
             json={"prompt": prompt, "max_tokens": 1000, "temperature": 0.15},
-            timeout=180
+            timeout=1000
         )
-
-        if response.status_code != 200:
+        if response.status_code == 200:
+            return response.json().get("response", "No entities extracted")
+        else:
             return f"[ERROR] API Error: {response.status_code} - {response.text}"
 
-        try:
-            return response.json().get("response", "No entities extracted")
-        except ValueError:
-            return f"[ERROR] Failed to parse model response as JSON:\n\n{response.text[:500]}"
 
     except requests.exceptions.Timeout:
         return "[ERROR] Entity extraction timeout - please try again"
@@ -373,12 +370,13 @@ Text: {search_results}"""
         response = requests.post(
             modal_endpoint,
             json={"prompt": prompt, "max_tokens": 1000, "temperature": 0.15},
-            timeout=1000  # Increased timeout
+            timeout=1000
         )
         if response.status_code == 200:
             return response.json().get("response", "No entities extracted")
         else:
             return f"[ERROR] API Error: {response.status_code} - {response.text}"
+
     except requests.exceptions.Timeout:
         return "[ERROR] Entity extraction timeout - please try again"
     except Exception as e: