Update Groq API keys with rotation logic

app.py

@@ -478,10 +478,15 @@ def generate_heatmap_image(data: np.ndarray, index_type: str, gaussian_sigma: fl
 
 # Groq API keys with cascading fallback
 GROQ_API_KEYS = [
+    "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",
     "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",
     "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",
     "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",
     "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",
+    "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",
+    "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",
+    "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",
+    "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"
 ]
 
 def run_llm_analysis(metric: str, stress_context: dict, indices_data: dict, 

llm_analysis.py

@@ -16,39 +16,72 @@ from typing import Dict, Any, List, Optional
 # GROQ API CONFIGURATION
 # ============================================================================
 
-GROQ_API_KEY = os.environ.get("GROQ_API_KEY", "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")
+# ============================================================================
+# GROQ API CONFIGURATION
+# ============================================================================
+
+GROQ_API_KEYS = [
+    "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",
+    "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",
+    "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",
+    "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",
+    "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",
+    "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",
+    "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",
+    "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",
+    "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"
+]
 GROQ_MODEL = "llama-3.3-70b-versatile"
 
 
 def call_groq(prompt: str) -> str:
-    """Call Groq API for LLM completion."""
+    """Call Groq API for LLM completion with Key Rotation."""
     from groq import Groq
     
-    if not GROQ_API_KEY:
-        raise ValueError("No GROQ_API_KEY configured")
-    
-    client = Groq(api_key=GROQ_API_KEY)
+    if not GROQ_API_KEYS:
+        raise ValueError("No GROQ_API_KEYS configured")
     
     max_attempts = 3
     base_backoff = 1.0
+    last_error = None
     
-    for attempt in range(1, max_attempts + 1):
+    # Iterate through all available keys
+    for key_idx, api_key in enumerate(GROQ_API_KEYS):
         try:
-            chat_completion = client.chat.completions.create(
-                messages=[
-                    {"role": "system", "content": "You are an expert agricultural AI. Respond with valid JSON only."},
-                    {"role": "user", "content": prompt}
-                ],
-                model=GROQ_MODEL,
-                temperature=0.7,
-                max_tokens=2048,
-            )
-            return chat_completion.choices[0].message.content
+            client = Groq(api_key=api_key)
+            print(f"[LLM Analysis] Trying key {key_idx+1}/{len(GROQ_API_KEYS)}")
+            
+            # Retry logic for EACH key (networking/timeout issues)
+            for attempt in range(1, max_attempts + 1):
+                try:
+                    chat_completion = client.chat.completions.create(
+                        messages=[
+                            {"role": "system", "content": "You are an expert agricultural AI. Respond with valid JSON only."},
+                            {"role": "user", "content": prompt}
+                        ],
+                        model=GROQ_MODEL,
+                        temperature=0.7,
+                        max_tokens=2048,
+                    )
+                    return chat_completion.choices[0].message.content
+                except Exception as e:
+                    # If it's a rate limit (429), break inner loop to try next key immediately
+                    if "429" in str(e) or "rate_limit" in str(e).lower():
+                        print(f"[LLM Analysis] Key {key_idx+1} rate limited. Switching to next key...")
+                        break 
+                    
+                    if attempt < max_attempts:
+                        time.sleep(base_backoff * (2 ** (attempt - 1)))
+                    else:
+                        raise e # Propagate to outer loop to try next key
+            
         except Exception as e:
-            if attempt < max_attempts:
-                time.sleep(base_backoff * (2 ** (attempt - 1)))
-            else:
-                raise ValueError(f"Groq API failed: {str(e)}")
+            last_error = str(e)
+            print(f"[LLM Analysis] Key {key_idx+1} failed: {e}")
+            continue
+            
+    # If all keys fail
+    raise ValueError(f"All {len(GROQ_API_KEYS)} Groq API keys failed. Last error: {last_error}")
     
     raise ValueError("All Groq API attempts failed")