Update app.py

app.py

@@ -220,56 +220,142 @@ def transcribe_audio(wav_file_path):
             os.remove(wav_file_path)
             print(f"Temporary WAV file deleted: {wav_file_path}")
 
+import time
+import requests
 
-def query_gemini_api(transcription):
+def query_gemini_api(transcription, max_retries=3, backoff_factor=2):
     """
     Send transcription text to Gemini API and fetch structured recipe information synchronously.
+    Includes automatic retries with exponential backoff for transient rate-limit errors (HTTP 429).
+    If all attempts fail, returns a default structured response.
     """
-    try:
-        # Define the structured prompt
-        prompt = (
-            "Analyze the provided cooking video transcription and extract the following structured information:\n"
-            "1. Recipe Name: Identify the name of the dish being prepared.\n"
-            "2. Ingredients List: Extract a detailed list of ingredients with their respective quantities (if mentioned).\n"
-            "3. Steps for Preparation: Provide a step-by-step breakdown of the recipe's preparation process, organized and numbered sequentially.\n"
-            "4. Cooking Techniques Used: Highlight the cooking techniques demonstrated in the video, such as searing, blitzing, wrapping, etc.\n"
-            "5. Equipment Needed: List all tools, appliances, or utensils mentioned, e.g., blender, hot pan, cling film, etc.\n"
-            "6. Nutritional Information (if inferred): Provide an approximate calorie count or nutritional breakdown based on the ingredients used.\n"
-            "7. Serving size: In count of people or portion size.\n"
-            "8. Special Notes or Variations: Include any specific tips, variations, or alternatives mentioned.\n"
-            "9. Festive or Thematic Relevance: Note if the recipe has any special relevance to holidays, events, or seasons.\n"
-            f"Text: {transcription}\n"
-        )
-
-        # Prepare the payload and headers
-        payload = {
-            "contents": [
-                {
-                    "parts": [
-                        {"text": prompt}
-                    ]
-                }
-            ]
-        }
-        headers = {"Content-Type": "application/json"}
-
-        # Send request to Gemini API synchronously
-        response = requests.post(
-            f"{GEMINI_API_ENDPOINT}?key={GEMINI_API_KEY}",
-            json=payload,
-            headers=headers,
-        )
-
-        # Raise error if response code is not 200
-        response.raise_for_status()
-
-        data = response.json()
-
-        return data.get("candidates", [{}])[0].get("content", {}).get("parts", [{}])[0].get("text", "No result found")
-
-    except requests.exceptions.RequestException as e:
-        print(f"Error querying Gemini API: {e}")
-        return {"error": str(e)}
+    # Define the structured prompt
+    prompt = (
+        "Analyze the provided cooking video transcription and extract the following structured information:\n"
+        "1. Recipe Name: Identify the name of the dish being prepared.\n"
+        "2. Ingredients List: Extract a detailed list of ingredients with their respective quantities (if mentioned).\n"
+        "3. Steps for Preparation: Provide a step-by-step breakdown of the recipe's preparation process, organized and numbered sequentially.\n"
+        "4. Cooking Techniques Used: Highlight the cooking techniques demonstrated in the video, such as searing, blitzing, wrapping, etc.\n"
+        "5. Equipment Needed: List all tools, appliances, or utensils mentioned, e.g., blender, hot pan, cling film, etc.\n"
+        "6. Nutritional Information (if inferred): Provide an approximate calorie count or nutritional breakdown based on the ingredients used.\n"
+        "7. Serving size: In count of people or portion size.\n"
+        "8. Special Notes or Variations: Include any specific tips, variations, or alternatives mentioned.\n"
+        "9. Festive or Thematic Relevance: Note if the recipe has any special relevance to holidays, events, or seasons.\n"
+        f"Text: {transcription}\n"
+    )
+
+    # Prepare the payload and headers
+    payload = {
+        "contents": [
+            {
+                "parts": [
+                    {"text": prompt}
+                ]
+            }
+        ]
+    }
+    headers = {"Content-Type": "application/json"}
+
+    # Attempt sending the request with retries
+    for attempt in range(max_retries):
+        try:
+            response = requests.post(
+                f"{GEMINI_API_ENDPOINT}?key={GEMINI_API_KEY}",
+                json=payload,
+                headers=headers,
+            )
+
+            # Check for 429 explicitly and wait before retrying
+            if response.status_code == 429:
+                if attempt < max_retries - 1:
+                    wait_time = backoff_factor ** attempt
+                    print(f"[Gemini] Rate limit reached (429). Retrying in {wait_time} seconds...")
+                    time.sleep(wait_time)
+                    continue
+                else:
+                    raise requests.exceptions.RequestException("Rate limit (429) after max retries")
+
+            response.raise_for_status()
+            data = response.json()
+            return data.get("candidates", [{}])[0].get("content", {}).get("parts", [{}])[0].get("text", "No result found")
+
+        except requests.exceptions.RequestException as e:
+            print(f"Error querying Gemini API: {e}")
+            if attempt < max_retries - 1:
+                wait_time = backoff_factor ** attempt
+                print(f"Retrying in {wait_time} seconds...")
+                time.sleep(wait_time)
+            else:
+                # If all retries fail, return a default structured response
+                default_response = (
+                    "Recipe Name: none\n"
+                    "Ingredients List: none\n"
+                    "Steps for Preparation: none\n"
+                    "Cooking Techniques Used: none\n"
+                    "Equipment Needed: none\n"
+                    "Nutritional Information: none\n"
+                    "Serving size: none\n"
+                    "Special Notes or Variations: none\n"
+                    "Festive or Thematic Relevance: none"
+                )
+                return default_response
+
+    # In case the loop finishes without returning, which should not happen
+    return {
+        "error": "Unexpected error occurred in query_gemini_api."
+    }
+
+
+
+# def query_gemini_api(transcription):
+#     """
+#     Send transcription text to Gemini API and fetch structured recipe information synchronously.
+#     """
+#     try:
+#         # Define the structured prompt
+#         prompt = (
+#             "Analyze the provided cooking video transcription and extract the following structured information:\n"
+#             "1. Recipe Name: Identify the name of the dish being prepared.\n"
+#             "2. Ingredients List: Extract a detailed list of ingredients with their respective quantities (if mentioned).\n"
+#             "3. Steps for Preparation: Provide a step-by-step breakdown of the recipe's preparation process, organized and numbered sequentially.\n"
+#             "4. Cooking Techniques Used: Highlight the cooking techniques demonstrated in the video, such as searing, blitzing, wrapping, etc.\n"
+#             "5. Equipment Needed: List all tools, appliances, or utensils mentioned, e.g., blender, hot pan, cling film, etc.\n"
+#             "6. Nutritional Information (if inferred): Provide an approximate calorie count or nutritional breakdown based on the ingredients used.\n"
+#             "7. Serving size: In count of people or portion size.\n"
+#             "8. Special Notes or Variations: Include any specific tips, variations, or alternatives mentioned.\n"
+#             "9. Festive or Thematic Relevance: Note if the recipe has any special relevance to holidays, events, or seasons.\n"
+#             f"Text: {transcription}\n"
+#         )
+
+#         # Prepare the payload and headers
+#         payload = {
+#             "contents": [
+#                 {
+#                     "parts": [
+#                         {"text": prompt}
+#                     ]
+#                 }
+#             ]
+#         }
+#         headers = {"Content-Type": "application/json"}
+
+#         # Send request to Gemini API synchronously
+#         response = requests.post(
+#             f"{GEMINI_API_ENDPOINT}?key={GEMINI_API_KEY}",
+#             json=payload,
+#             headers=headers,
+#         )
+
+#         # Raise error if response code is not 200
+#         response.raise_for_status()
+
+#         data = response.json()
+
+#         return data.get("candidates", [{}])[0].get("content", {}).get("parts", [{}])[0].get("text", "No result found")
+
+#     except requests.exceptions.RequestException as e:
+#         print(f"Error querying Gemini API: {e}")
+#         return {"error": str(e)}
 
 
 if __name__ == '__main__':