Update process_interview.py

process_interview.py

@@ -50,18 +50,24 @@ PINECONE_KEY = os.getenv("PINECONE_KEY")
 ASSEMBLYAI_KEY = os.getenv("ASSEMBLYAI_KEY")
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 
-def download_audio_from_url(url: str) -> str:
-    """Downloads an audio file from a URL to a temporary local path."""
+def download_audio_from_url(url: str, retries=3) -> str:
+    """Downloads an audio file from a URL to a temporary local path with retries."""
     try:
         temp_dir = tempfile.gettempdir()
         temp_path = os.path.join(temp_dir, f"{uuid.uuid4()}.tmp_audio")
         logger.info(f"Downloading audio from {url} to {temp_path}")
-        with requests.get(url, stream=True) as r:
-            r.raise_for_status()
-            with open(temp_path, 'wb') as f:
-                for chunk in r.iter_content(chunk_size=8192):
-                    f.write(chunk)
-        return temp_path
+        for attempt in range(retries):
+            try:
+                with requests.get(url, stream=True, timeout=30) as r:
+                    r.raise_for_status()
+                    with open(temp_path, 'wb') as f:
+                        for chunk in r.iter_content(chunk_size=8192):
+                            f.write(chunk)
+                return temp_path
+            except (requests.exceptions.ChunkedEncodingError, urllib3.exceptions.ProtocolError) as e:
+                logger.warning(f"Attempt {attempt + 1} failed: {e}. Retrying...")
+                time.sleep(2 ** attempt) # Exponential backoff
+        raise Exception(f"Failed to download audio after {retries} attempts.")
     except Exception as e:
         logger.error(f"Failed to download audio from URL {url}: {e}")
         raise