Update app.py

app.py

@@ -412,13 +412,10 @@ def run_production_pipeline(job: TranslationJob, gemini_key: str, elevenlabs_key
         run_ffmpeg_command(cmd_extract, "Audio Extraction")
         
         # 3. Request Transcription & Translation from Gemini 2.5 Flash
-        print(f"[{video_id}] Calling Gemini 2.5 Flash for transcription/translation to '{job.target_lang}'...")
+        print(f"[{video_id}] Calling Gemini for transcription/translation to '{job.target_lang}'...")
         with open(extracted_audio_path, "rb") as audio_file:
             audio_data = base64.b64encode(audio_file.read()).decode("utf-8")
             
-        # 🎯 DÜZELTME 3: Endpoint URL'i v1beta olarak bırakıldı, faturalı bakiye ile limitsiz akacak!
-        gemini_url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={gemini_key}"
-        
         if job.has_captions:
             gemini_prompt = f"Transcribe the following audio, translate it accurately to '{job.target_lang}', and output the result in SRT subtitle format. Keep each caption line short (max 4-5 words) and ensure the timing is aligned with the audio. Output ONLY the raw SRT text, no extra markdown formatting, tags, or explanations."
         else:
@@ -441,9 +438,45 @@ def run_production_pipeline(job: TranslationJob, gemini_key: str, elevenlabs_key
                 }
             ]
         }
-        gemini_res = requests.post(gemini_url, json=gemini_payload)
-        gemini_res.raise_for_status()
-        gemini_data = gemini_res.json()
+        
+        # Retry logic: try gemini-2.5-flash up to 3 times (503/429/500 are transient),
+        # then fall back to gemini-2.0-flash-exp if it keeps failing.
+        GEMINI_MODELS = [
+            "gemini-2.5-flash",       # Primary: best quality
+            "gemini-2.0-flash-exp",   # Fallback: still very capable
+        ]
+        RETRY_DELAYS = [5, 15, 30]   # seconds between retries per model
+        
+        gemini_data = None
+        last_error = None
+        
+        for model_name in GEMINI_MODELS:
+            gemini_url = f"https://generativelanguage.googleapis.com/v1beta/models/{model_name}:generateContent?key={gemini_key}"
+            print(f"[{video_id}] Trying Gemini model: {model_name}")
+            
+            for attempt, delay in enumerate(RETRY_DELAYS, start=1):
+                try:
+                    gemini_res = requests.post(gemini_url, json=gemini_payload, timeout=300)
+                    if gemini_res.status_code in (503, 429, 500):
+                        last_error = f"HTTP {gemini_res.status_code}"
+                        print(f"[{video_id}] Gemini {model_name} attempt {attempt} returned {gemini_res.status_code}. Retrying in {delay}s...")
+                        time.sleep(delay)
+                        continue
+                    gemini_res.raise_for_status()
+                    gemini_data = gemini_res.json()
+                    print(f"[{video_id}] Gemini {model_name} responded successfully on attempt {attempt}.")
+                    break  # success
+                except Exception as e:
+                    last_error = str(e)
+                    print(f"[{video_id}] Gemini {model_name} attempt {attempt} failed: {e}. Retrying in {delay}s...")
+                    time.sleep(delay)
+            
+            if gemini_data:
+                break  # got a response, no need to try next model
+            print(f"[{video_id}] All retries exhausted for {model_name}, trying next model...")
+        
+        if not gemini_data:
+            raise Exception(f"Gemini API failed after all retries and model fallbacks. Last error: {last_error}")
         
         gemini_output = gemini_data["candidates"][0]["content"]["parts"][0]["text"].strip()
         
@@ -670,10 +703,10 @@ def run_production_pipeline(job: TranslationJob, gemini_key: str, elevenlabs_key
             }
             upload_res = requests.post(storage_upload_url, headers=upload_headers, data=out_file)
             # If exists, we can try to PUT (overwrite)
-            if upload_res.status_code == 400 and "AlreadyExists" in upload_res.text:
+        if upload_res.status_code == 400 and "AlreadyExists" in upload_res.text:
                 upload_res = requests.put(storage_upload_url, headers=upload_headers, data=out_file)
-            
-            upload_res.raise_for_status()
+        
+        upload_res.raise_for_status()
             
         # Get public url
         public_video_url = f"{base_url}/storage/v1/object/public/videos/{video_id}_translated.mp4"