Update app.py

app.py

@@ -11,6 +11,7 @@ import gc
 import math
 import uuid
 import re
+import time
 from urllib.parse import quote
 from datetime import timedelta
 
@@ -257,7 +258,7 @@ def call_gemini_for_edl(transcript_data, story_prompt, api_key):
         st.error("Gemini API Key is missing.")
         return None
         
-    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-pro:generateContent?key={api_key}"
+    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-3.0-flash:generateContent?key={api_key}"
     
     system_prompt = (
         "You are an expert Documentary Senior Editor. Use the provided transcript JSON "
@@ -274,22 +275,69 @@ def call_gemini_for_edl(transcript_data, story_prompt, api_key):
     
     prompt_text = f"Creative Brief: {story_prompt}\n\nTranscript Data:\n{json.dumps(transcript_data)}"
     
+    # Stricter payload formatting according to Gemini REST API specs
     payload = {
-        "contents": [{"parts": [{"text": prompt_text}]}],
-        "systemInstruction": {"parts": [{"text": system_prompt}]},
-        "generationConfig": {"responseMimeType": "application/json"}
+        "contents": [
+            {
+                "role": "user",
+                "parts": [{"text": prompt_text}]
+            }
+        ],
+        "systemInstruction": {
+            "parts": [{"text": system_prompt}]
+        },
+        "generationConfig": {
+            "responseMimeType": "application/json"
+        }
     }
     
-    try:
-        res = requests.post(url, json=payload)
-        res.raise_for_status()
-        result_json = res.json()
-        raw_text = result_json['candidates'][0]['content']['parts'][0]['text']
-        cleaned_text = clean_json_response(raw_text)
-        return json.loads(cleaned_text)
-    except Exception as e:
-        st.error(f"Senior Editor AI Error: {e}")
-        return None
+    headers = {
+        "Content-Type": "application/json"
+    }
+    
+    max_retries = 3
+    for attempt in range(max_retries):
+        try:
+            res = requests.post(url, json=payload, headers=headers)
+            
+            # Handle Rate Limiting (429) specifically
+            if res.status_code == 429:
+                if attempt < max_retries - 1:
+                    wait_time = 10 * (attempt + 1)
+                    st.warning(f"Rate limit hit. Retrying in {wait_time} seconds...")
+                    time.sleep(wait_time)
+                    continue
+                else:
+                    st.error("Google API Rate Limit Exceeded (429). Please wait a few minutes before trying again, or upgrade your API tier.")
+                    return None
+            
+            # Instead of generic failure, show the EXACT error message from Google
+            if not res.ok:
+                error_details = res.text
+                try:
+                    error_details = res.json().get('error', {}).get('message', res.text)
+                except Exception:
+                    pass
+                st.error(f"Google API Rejected Request ({res.status_code}): {error_details}")
+                return None
+                
+            result_json = res.json()
+            raw_text = result_json['candidates'][0]['content']['parts'][0]['text']
+            cleaned_text = clean_json_response(raw_text)
+            return json.loads(cleaned_text)
+            
+        except json.JSONDecodeError as e:
+            st.error(f"Failed to parse Gemini's output as JSON: {e}")
+            return None
+        except requests.exceptions.RequestException as e:
+            if attempt < max_retries - 1:
+                time.sleep(5)
+                continue
+            st.error(f"Network Error connecting to Gemini API: {e}")
+            return None
+        except Exception as e:
+            st.error(f"Senior Editor AI Error: {e}")
+            return None
 
 # --- Streamlit UI ---
 st.set_page_config(page_title="Junior Editor", layout="wide")