The Gemini call now retries up to 3 times with a backoff delay (2s, then 4s) between attempts.

1 files changed, 31 insertions, 25 deletions

diff --git a/server.py b/server.py
index a290c80..033b080 100755
--- a/server.py
+++ b/server.py
@@ -174,33 +174,39 @@ def classify_stress_with_gemini(sentence_text):
         print("Stress prompt template is not available. Skipping classification.")
         return "error_no_prompt_template"
 
-    try:
-        # Call the API using the new Client
-        prompt_text = STRESS_PROMPT_TEMPLATE + f'\nSentence: {sentence_text}\nClassification:'
-        print(f"Gemini API call: few-shot stress classification (gemini-2.5-pro)\n  Prompt: {prompt_text}")
-        response = gemini_client.models.generate_content(
-            model='gemini-2.5-pro',
-            contents=prompt_text
-        )
-        print(f"  Gemini response: {response.text.strip()}")
-        # We need to parse the classification from the response
-        classification = response.text.strip().lower()
+    import time
+    prompt_text = STRESS_PROMPT_TEMPLATE + f'\nSentence: {sentence_text}\nClassification:'
+    last_error = None
+    for attempt in range(3):
+        try:
+            if attempt > 0:
+                time.sleep(2 * attempt)
+                print(f"  Gemini retry {attempt + 1}/3")
+            print(f"Gemini API call: few-shot stress classification (gemini-2.5-pro)\n  Prompt: {prompt_text}")
+            response = gemini_client.models.generate_content(
+                model='gemini-2.5-pro',
+                contents=prompt_text
+            )
+            print(f"  Gemini response: {response.text.strip()}")
+            classification = response.text.strip().lower()
+
+            if "cellular" in classification:
+                result = "neg"  # 'neg' for Cellular Level Stress
+            elif "organismal" in classification:
+                result = "pos"  # 'pos' for Organismal Stress
+            else:
+                print(f"Warning: Gemini returned unexpected classification: '{classification}' for sentence: '{sentence_text}'")
+                result = "unknown"
+            if result in ("pos", "neg"):
+                _gemini_cache[cache_key] = result
+            return result
 
-        # The model might return "Cellular Level Stress" or "Organismal Stress"
-        if "cellular" in classification:
-            result = "neg"  # 'neg' for Cellular Level Stress
-        elif "organismal" in classification:
-            result = "pos"  # 'pos' for Organismal Stress
-        else:
-            print(f"Warning: Gemini returned unexpected classification: '{classification}' for sentence: '{sentence_text}'")
-            result = "unknown"
-        if result in ("pos", "neg"):
-            _gemini_cache[cache_key] = result
-        return result
+        except Exception as e:
+            last_error = e
+            print(f"Error calling Gemini API (attempt {attempt + 1}/3): {e}")
 
-    except Exception as e:
-        print(f"Error calling Gemini API for stress classification: {e}")
-        return "error_api_call"
+    print(f"Gemini API failed after 3 attempts: {last_error}")
+    return "error_api_call"
 
 
 # zero-shot Function to classify stress using Gemini API