Update main.py

main.py

@@ -1,4 +1,4 @@
-# main.py (Full Code - Specific April 2025 Models: Llama 4 Scout & DeepSeek V3 Free)
+# main.py (Full Code - Specific April 2025 Models: Llama 4 Scout & DeepSeek V3 Free - FIXED)
 import os
 import re
 import logging
@@ -536,7 +536,8 @@ async def _call_groq(text: str, summary_type: str) -> Tuple[Optional[str], Optio
     try:
         groq_client = Groq( api_key=GROQ_API_KEY, timeout=httpx.Timeout(120.0, connect=10.0) )
         logger.info(f"[Groq Primary] Sending request to Groq ({GROQ_LLAMA4_MODEL})...")
-        chat_completion = await groq_client.chat.completions.create(
+        # FIX: Removed await from the synchronous call
+        chat_completion = groq_client.chat.completions.create(
             messages=[ { "role": "user", "content": full_prompt } ],
             model=GROQ_LLAMA4_MODEL,
             temperature=0.7, # Adjust from Groq default of 1 if needed
@@ -567,11 +568,18 @@ async def _call_groq(text: str, summary_type: str) -> Tuple[Optional[str], Optio
          return None, f"Sorry, couldn't connect to the primary AI service ({GROQ_LLAMA4_MODEL})."
     except Exception as e:
         logger.error(f"[Groq Primary] Unexpected error during Groq API call: {e}", exc_info=True);
+        # Don't expose internal TypeError potentially, give a generic message
+        # Check if the specific error is the TypeError we saw earlier.
+        if isinstance(e, TypeError) and "can't be used in 'await' expression" in str(e):
+             # This case *shouldn't* happen now await is removed, but good to handle defensively.
+             logger.error("[Groq Primary] Encountered unexpected await TypeError again.")
+             return None, f"Sorry, an internal configuration error occurred with the primary AI service ({GROQ_LLAMA4_MODEL})."
         return None, f"Sorry, an unexpected error occurred while using the primary AI service ({GROQ_LLAMA4_MODEL})."
 
+
 async def _call_gemini(text: str, summary_type: str, model_name: str) -> Tuple[Optional[str], Optional[str]]:
     """Internal function to call Gemini API. Returns (summary, error_message)."""
-    global _gemini_api_enabled
+    global _gemini_api_enabled, HarmCategory, HarmBlockThreshold # Ensure HarmCategory/Threshold are available if SDK is loaded
     if not _gemini_api_enabled:
         logger.error(f"[Gemini {model_name}] Called but API is disabled.");
         return None, f"Error: AI service (Gemini API) not configured/available."
@@ -583,11 +591,23 @@ async def _call_gemini(text: str, summary_type: str, model_name: str) -> Tuple[O
         logger.warning(f"[Gemini {model_name}] Input length ({len(text)}) exceeds limit ({MAX_INPUT_LENGTH_GEMINI}). Truncating.");
         text = text[:MAX_INPUT_LENGTH_GEMINI] + "... (Content truncated)"
     full_prompt = f"{prompt}\n\n{text}"
-    safety_settings = { HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE, HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE, HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE, HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE, }
-    for category_name in dir(HarmCategory):
-        if category_name.startswith('HARM_CATEGORY_') and getattr(HarmCategory, category_name) not in safety_settings:
-            safety_settings[getattr(HarmCategory, category_name)] = HarmBlockThreshold.BLOCK_NONE
-    logger.debug(f"[Gemini {model_name}] Using safety settings: { {k.name: v.name for k, v in safety_settings.items()} }")
+
+    # FIX: Define safety_settings explicitly with the 5 categories from the error log
+    # Removed the problematic loop that dynamically added potentially unsupported categories.
+    try:
+        safety_settings = {
+            HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY: HarmBlockThreshold.BLOCK_NONE, # Added based on error msg
+        }
+        logger.debug(f"[Gemini {model_name}] Using safety settings: { {k.name: v.name for k, v in safety_settings.items()} }")
+    except AttributeError:
+        # Handle case where HarmCategory or HarmBlockThreshold might not be loaded correctly
+        # This shouldn't happen if _gemini_sdk_available is True, but defensive coding
+        logger.error(f"[Gemini {model_name}] Failed to define safety settings. HarmCategory/HarmBlockThreshold missing?")
+        return None, f"Sorry, an internal error occurred configuring the AI service ({model_name})."
 
     try:
         logger.debug(f"[Gemini {model_name}] Initializing model {model_name}")
@@ -629,9 +649,14 @@ async def _call_gemini(text: str, summary_type: str, model_name: str) -> Tuple[O
          return None, f"Sorry, there was an issue processing the response from the AI service ({model_name})."
     except Exception as e:
         logger.error(f"[Gemini {model_name}] Unexpected error during Gemini API call: {e}", exc_info=True);
+        # Check if it's the specific InvalidArgument error we were seeing
+        if isinstance(e, google.api_core.exceptions.InvalidArgument) and "safety_settings" in str(e):
+             logger.error(f"[Gemini {model_name}] Safety settings InvalidArgument error persisted: {e}")
+             return None, f"Sorry, the AI service ({model_name}) reported an invalid configuration (Safety Settings). Please report this issue."
         error_msg = f"Sorry, an unexpected error occurred while using the AI service ({model_name})."
         return None, error_msg
 
+
 async def _call_openrouter(text: str, summary_type: str) -> Tuple[Optional[str], Optional[str]]:
     """Internal function to call OpenRouter API (Final Fallback - DeepSeek V3 Free). Returns (summary, error_message)."""
     global OPENROUTER_API_KEY, OPENROUTER_DEEPSEEK_MODEL, _openrouter_fallback_enabled
@@ -741,7 +766,7 @@ async def generate_summary(text: str, summary_type: str) -> str:
         errors["GeminiFlash"] = "Service disabled/unavailable."
 
     # --- Attempt 3: Gemini 2.0 Flash ---
-    if _gemini_api_enabled and errors["GeminiFlash"] is None:
+    if _gemini_api_enabled and errors["GeminiFlash"] is None: # Check if Gemini API is enabled AND wasn't already marked as failed/skipped
         logger.info(f"[Summary Generation] Attempting 3: Gemini ({GEMINI_FLASH_MODEL})")
         summary, errors["GeminiFlash"] = await _call_gemini(text, summary_type, GEMINI_FLASH_MODEL)
         if summary:
@@ -749,7 +774,7 @@ async def generate_summary(text: str, summary_type: str) -> str:
             return summary
         else:
             logger.warning(f"[Summary Generation] Gemini 2.0 Flash failed. Error: {errors['GeminiFlash']}. Proceeding to OpenRouter DeepSeek V3.")
-    elif errors["GeminiFlash"] is None:
+    elif errors["GeminiFlash"] is None: # Only log skip message if it wasn't already marked as disabled
          logger.warning("[Summary Generation] Skipping Gemini 2.0 Flash (API was disabled).")
          errors["GeminiFlash"] = "Service disabled/unavailable."
 
@@ -1106,7 +1131,7 @@ async def health_check(request: Request) -> PlainTextResponse:
 
 async def telegram_webhook(request: Request) -> Response:
     """Handles incoming updates from Telegram."""
-    global WEBHOOK_SECRET
+    global WEBHOOK_SECRET, ptb_app # Ensure ptb_app is accessible
     if not ptb_app: logger.error("Webhook received but PTB application not initialized."); return PlainTextResponse('Bot not initialized', status_code=503)
     if not ptb_app.running: logger.warning("Webhook received but PTB application not running."); return PlainTextResponse('Bot not running, cannot process update', status_code=503)
     if WEBHOOK_SECRET:
@@ -1117,7 +1142,7 @@ async def telegram_webhook(request: Request) -> Response:
         logger.debug(f"Processing update_id: {update.update_id} via webhook"); await ptb_app.process_update(update)
         return Response(status_code=200)
     except json.JSONDecodeError: logger.error("Webhook received invalid JSON."); return PlainTextResponse('Bad Request: Invalid JSON', status_code=400)
-    except Exception as e: logger.error(f"Error processing webhook update: {e}", exc_info=True); return Response(status_code=200)
+    except Exception as e: logger.error(f"Error processing webhook update: {e}", exc_info=True); return Response(status_code=200) # Return 200 to TG even if processing fails, to avoid retries
 
 # --- Starlette App Definition ---
 app = Starlette(
@@ -1133,6 +1158,12 @@ logger.info("Starlette ASGI application created with health check and webhook ro
 # --- Development Server (if run directly) ---
 if __name__ == '__main__':
     import uvicorn
+    # Need to import google.api_core.exceptions for the explicit check in _call_gemini error handling
+    try:
+        import google.api_core.exceptions
+    except ImportError:
+        pass # It's already handled by _gemini_sdk_available flag
+
     logger.warning("Running in development mode using Uvicorn directly - NOT for production!")
     log_level = os.environ.get("LOGGING_LEVEL", "info").lower()
     local_port = int(os.environ.get('PORT', 8080))