timeout increased

app.py

@@ -217,25 +217,25 @@ def send_input(user_id):
         return response.ok
 
 def run_fhe(user_id):
-    """Apply the seizure detection model on the encrypted image previously sent using FHE.
-
-    Args:
-        user_id (int): The current user's ID.
-    """
-    data = {
-        "user_id": user_id,
-    }
-
-    # Trigger the FHE execution on the encrypted image previously sent
+    """Apply the seizure detection model on the encrypted image previously sent using FHE."""
+    data = {"user_id": user_id}
     url = SERVER_URL + "run_fhe"
-    with requests.post(
-        url=url,
-        data=data,
-    ) as response:
-        if response.ok:
-            return response.json()
-        else:
-            raise gr.Error("Please wait for the input image to be sent to the server.")
+    
+    try:
+        with requests.post(url=url, data=data, timeout=300) as response:
+            response.raise_for_status()  # Raises an HTTPError for bad responses
+            if response.ok:
+                return response.json()
+            else:
+                raise gr.Error(f"Server responded with status code {response.status_code}")
+    except requests.exceptions.Timeout:
+        raise gr.Error("The request timed out. The server might be overloaded.")
+    except requests.exceptions.ConnectionError:
+        raise gr.Error("Failed to connect to the server. Please check your network connection.")
+    except requests.exceptions.RequestException as e:
+        raise gr.Error(f"An error occurred: {str(e)}")
+    except Exception as e:
+        raise gr.Error(f"An unexpected error occurred: {str(e)}")
 
 def get_output(user_id):
     """Retrieve the encrypted output (boolean).