polling

app.py

@@ -245,6 +245,8 @@ def send_input(user_id):
     with requests.post(url=url, data=data, files=files) as response:
         return response.ok
 
+import time
+
 def run_fhe(user_id):
     """Apply the seizure detection model on the encrypted image previously sent using FHE."""
     data = {"user_id": user_id}
@@ -252,20 +254,23 @@ def run_fhe(user_id):
     
     try:
         logger.info(f"Sending request to {url} with user_id: {user_id}")
-        with requests_retry_session().post(url=url, data=data, timeout=300) as response:
-            logger.info(f"Received response with status code: {response.status_code}")
-            response.raise_for_status()  # Raises an HTTPError for bad responses
-            if response.ok:
-                return response.json()
-            else:
-                logger.error(f"Server responded with status code {response.status_code}")
-                raise gr.Error(f"Server responded with status code {response.status_code}")
-    except requests.exceptions.Timeout:
-        logger.error("The request timed out. The server might be overloaded.")
-        raise gr.Error("The request timed out. The server might be overloaded.")
-    except requests.exceptions.ConnectionError as e:
-        logger.error(f"Failed to connect to the server. Error: {str(e)}")
-        raise gr.Error("Failed to connect to the server. Please check your network connection.")
+        
+        max_attempts = 20
+        attempt = 0
+        while attempt < max_attempts:
+            try:
+                with requests_retry_session().post(url=url, data=data, timeout=30) as response:
+                    response.raise_for_status()
+                    result = response.json()
+                    logger.info(f"FHE execution completed. Result: {result}")
+                    return result.get('execution_time', 'N/A')
+            except requests.exceptions.Timeout:
+                logger.info(f"Request timed out. Retrying... (Attempt {attempt + 1}/{max_attempts})")
+                time.sleep(15)  # Wait for 15 seconds before retrying
+                attempt += 1
+        
+        raise gr.Error("FHE execution timed out after multiple attempts")
+    
     except requests.exceptions.RequestException as e:
         logger.error(f"An error occurred: {str(e)}")
         raise gr.Error(f"An error occurred: {str(e)}")