Update Groq API keys with rotation logic

app.py

@@ -52,15 +52,21 @@ SENTINEL2_API_URL = os.getenv("SENTINEL2_API_URL", "https://aniket2006-agrow-sen
 # ============================================================================
 # GROQ SETUP
 # ============================================================================
-GROQ_API_KEY = os.environ.get("GROQ_API_KEY", "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")
+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"
 
-if GROQ_API_KEY:
-    groq_client = Groq(api_key=GROQ_API_KEY)
-    logger.info(f"Groq API configured with model {GROQ_MODEL}")
-else:
-    groq_client = None
-    logger.warning("GROQ_API_KEY not set - chatbot will return mock responses")
+# Verify keys loaded
+logger.info(f"Loaded {len(GROQ_API_KEYS)} Groq API keys")
 
 # Supabase
 supabase = SupabaseClient()
@@ -500,39 +506,52 @@ Your analysis:"""
 # GENERATE RESPONSE
 # ============================================================================
 def generate_response(user_message: str, history: List[Dict], context: Dict) -> tuple[str, List[str]]:
-    """Generate AI response using comprehensive context."""
+    """Generate AI response using comprehensive context with API Key Rotation."""
     context = context or {}
     history = history or []
     context_used = context.get("data_sources", [])
     
-    if groq_client is None:
-        return "Please configure GROQ_API_KEY for real responses.", []
-    
-    try:
-        prompt = build_llm_prompt(user_message, context, history)
-        
-        chat_completion = groq_client.chat.completions.create(
-            messages=[
-                {
-                    "role": "system",
-                    "content": "You are AGROW AI, an expert agricultural advisor. Provide helpful, data-driven advice."
-                },
-                {
-                    "role": "user",
-                    "content": prompt
-                }
-            ],
-            model=GROQ_MODEL,
-            temperature=0.7,
-            max_tokens=4096,
-        )
-        
-        return chat_completion.choices[0].message.content, context_used
-        
-    except Exception as e:
-        logger.error(f"Groq error: {e}")
-        traceback.print_exc()
-        return f"I apologize, but I encountered an error: {str(e)}", []
+    prompt = build_llm_prompt(user_message, context, history)
+    last_error = None
+
+    # Try keys sequentially with fallback on failure
+    for i, api_key in enumerate(GROQ_API_KEYS):
+        try:
+            logger.info(f"[Chatbot] Trying Groq API key {i+1}/{len(GROQ_API_KEYS)}")
+            client = Groq(api_key=api_key)
+            
+            chat_completion = client.chat.completions.create(
+                messages=[
+                    {
+                        "role": "system",
+                        "content": "You are AGROW AI, an expert agricultural advisor. Provide helpful, data-driven advice."
+                    },
+                    {
+                        "role": "user",
+                        "content": prompt
+                    }
+                ],
+                model=GROQ_MODEL,
+                temperature=0.7,
+                max_tokens=4096,
+            )
+            
+            return chat_completion.choices[0].message.content, context_used
+            
+        except Exception as e:
+            last_error = str(e)
+            logger.warning(f"[Chatbot] Key {i+1} failed: {e}")
+            # If it's not a rate limit issue, maybe we shouldn't retry? 
+            # But for robustness, we'll assume any error warrants trying another key.
+            continue
+            
+    # All keys failed
+    logger.error(f"[Chatbot] All {len(GROQ_API_KEYS)} keys failed. Last error: {last_error}")
+    traceback.print_exc()
+    fallback_msg = "I apologize, but I'm currently experiencing high traffic. Please try again in a moment."
+    if context.get('weather'): # Simple fallback if we have context
+        fallback_msg += f" (Weather: {context['weather'].get('current_temp')}°C)"
+    return fallback_msg, []
 
 
 # ============================================================================
@@ -548,7 +567,7 @@ async def root():
 
 @app.get("/health")
 async def health():
-    return {"status": "healthy", "groq_configured": groq_client is not None}
+    return {"status": "healthy", "groq_configured": len(GROQ_API_KEYS) > 0}
 
 
 @app.post("/session/new", response_model=SessionResponse)