Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 from moviepy.editor import VideoFileClip
 from pydub import AudioSegment
 from pydub.utils import make_chunks
+import time
 
 app = Flask(__name__)
 
@@ -52,19 +53,29 @@ def extract_audio(video_path):
         raise RuntimeError(f"Error extracting audio: {e}")
     return audio_path
 
-def transcribe_audio_chunk(audio_chunk_path):
-    try:
-        with open(audio_chunk_path, "rb") as audio_chunk:
-            headers = {"Authorization": f"Bearer {API_TOKEN}"}
-            response = requests.post(API_URL, headers=headers, files={"file": audio_chunk})
-
-        if response.status_code != 200:
-            raise RuntimeError(f"Error during transcription: {response.text}")
-
-        result = response.json()
-        return result.get("text", "")
-    except Exception as e:
-        raise RuntimeError(f"Error during transcription: {e}")
+def transcribe_audio_chunk(audio_chunk_path, retries=5, wait_time=60):
+    headers = {"Authorization": f"Bearer {API_TOKEN}"}
+    
+    for attempt in range(retries):
+        try:
+            with open(audio_chunk_path, "rb") as audio_chunk:
+                response = requests.post(API_URL, headers=headers, files={"file": audio_chunk})
+            
+            if response.status_code == 200:
+                result = response.json()
+                return result.get("text", "")
+            elif response.status_code == 503 and 'estimated_time' in response.json():
+                # If the model is loading, wait and retry
+                wait = response.json()['estimated_time']
+                time.sleep(wait)
+            else:
+                response.raise_for_status()
+        except Exception as e:
+            if attempt == retries - 1:
+                raise RuntimeError(f"Error during transcription after {retries} attempts: {e}")
+            else:
+                time.sleep(wait_time)
+    return ""
 
 def split_and_transcribe_audio(audio_path):
     if not os.path.exists(audio_path):