OpenAI Anthropic Gemini works

app.py

@@ -157,7 +157,8 @@ async def ask_gemini(query: str, history: List[Dict[str, str]]) -> str:
     }
 
     try:
-        async with httpx.AsyncClient() as client:
+        async with httpx.AsyncClient(timeout=30.0) as client:
+            logger.info(f"Sending Gemini request: {payload}")
             response = await client.post(
                 f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={gemini_api_key}",
                 headers=headers,
@@ -165,12 +166,36 @@ async def ask_gemini(query: str, history: List[Dict[str, str]]) -> str:
             )
         
         response.raise_for_status()
-        return response.json()['candidates'][0]['content']['parts'][0]['text']
+        response_json = response.json()
+        logger.info(f"Gemini response: {response_json}")
+        
+        # Validate response structure
+        if not isinstance(response_json, dict) or "candidates" not in response_json or not response_json["candidates"]:
+            logger.error(f"Invalid Gemini response structure: {response_json}")
+            return f"Error: Invalid Gemini response structure"
+        
+        candidates = response_json["candidates"]
+        if not isinstance(candidates, list) or not candidates or "content" not in candidates[0]:
+            logger.error(f"Invalid Gemini candidates format: {candidates}")
+            return f"Error: Invalid Gemini candidates format"
+        
+        content = candidates[0]["content"]
+        if not isinstance(content, dict) or "parts" not in content or not content["parts"]:
+            logger.error(f"Invalid Gemini content format: {content}")
+            return f"Error: Invalid Gemini content format"
+        
+        parts = content["parts"]
+        if not isinstance(parts, list) or not parts or "text" not in parts[0]:
+            logger.error(f"Invalid Gemini parts format: {parts}")
+            return f"Error: Invalid Gemini parts format"
+        
+        return parts[0]["text"]
     except httpx.HTTPStatusError as e:
-        logger.error(f"Gemini HTTP Status Error: {e.response.status_code}, {e.response.text}")
-        return f"Error: Gemini HTTP Status Error: {e.response.status_code}, {e.response.text}"
+        response_text = await e.response.aread()
+        logger.error(f"Gemini HTTP Status Error: {e.response.status_code}, {response_text.decode('utf-8')}")
+        return f"Error: Gemini HTTP Status Error: {e.response.status_code}, {response_text.decode('utf-8')}"
     except Exception as e:
-        logger.error(f"Gemini Error: {str(e)}")
+        logger.error(f"Gemini Error: {str(e)}\nStack trace: {traceback.format_exc()}")
         return f"Error: Gemini Error: {str(e)}"
 
 async def query_model(query: str, providers: List[str], history: List[Dict[str, str]]) -> AsyncGenerator[Tuple[str, List[Dict[str, str]]], None]:
@@ -203,7 +228,7 @@ async def query_model(query: str, providers: List[str], history: List[Dict[str,
     # Handle Gemini (non-streaming)
     if "Gemini" in providers:
         response = await ask_gemini(query, history)
-        if response.strip():
+        if response.strip() and not response.startswith("Error:"):
             responses.append(f"[Gemini]: {response}")
 
     # Combine responses