Update main.py

main.py

@@ -1,4 +1,4 @@
-# main.py (Updating Gemini Model and Safety Settings)
+# main.py (Correcting SyntaxError at line 424)
 import os
 import re
 import logging
@@ -89,7 +89,6 @@ GEMINI_API_KEY = get_secret('GEMINI_API_KEY')
 
 OPENROUTER_MODEL = os.environ.get("OPENROUTER_MODEL", "deepseek/deepseek-chat-v3-0324:free")
 APIFY_ACTOR_ID = os.environ.get("APIFY_ACTOR_ID", "karamelo~youtube-transcripts")
-# *** FIX: Update default Gemini model to gemini-2.0-flash ***
 GEMINI_MODEL = os.environ.get("GEMINI_MODEL", "gemini-2.0-flash")
 
 if not TELEGRAM_TOKEN: logger.critical("❌ FATAL: TELEGRAM_TOKEN not found."); raise RuntimeError("Exiting: Telegram token missing.")
@@ -163,7 +162,7 @@ async def get_transcript_via_supadata(video_id: str, api_key: str) -> Optional[s
     if not video_id: logger.error("[Supadata] No video_id provided"); return None
     if not api_key: logger.error("[Supadata] API key missing."); return None
     logger.info(f"[Supadata] Attempting fetch for video ID: {video_id}")
-    api_endpoint = "https://api.supadata.ai/v1/youtube/transcript"
+    api_endpoint = "https://api.supadata.ai/v1/youtube/transcript" # Corrected URL
     params = {"videoId": video_id, "format": "text"}; headers = {"X-API-Key": api_key}
     try:
         async with httpx.AsyncClient(timeout=30.0) as client:
@@ -334,23 +333,18 @@ async def generate_summary_gemini(text: str, summary_type: str) -> str:
     MAX_INPUT_LENGTH = 1000000
     if len(text) > MAX_INPUT_LENGTH: logger.warning(f"[Gemini Fallback] Input length ({len(text)}) exceeds limit ({MAX_INPUT_LENGTH}). Truncating."); text = text[:MAX_INPUT_LENGTH] + "... (Content truncated)"
     full_prompt = f"{prompt}\n\n{text}"
-
     # *** FIX: Set all safety settings to BLOCK_NONE ***
     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,
-        # Note: CIVIC_INTEGRITY might not be available/needed for all models,
-        # but setting it won't hurt if the HarmCategory exists.
-        # If it causes an error later, remove this line.
+        # getattr used for safety in case the category doesn't exist in the SDK version
         getattr(HarmCategory, 'HARM_CATEGORY_CIVIC_INTEGRITY', None): HarmBlockThreshold.BLOCK_NONE
     }
-    # Filter out None keys in case CIVIC_INTEGRITY doesn't exist
     safety_settings = {k: v for k, v in safety_settings.items() if k is not None}
     logger.debug(f"[Gemini Fallback] Using safety settings: {safety_settings}")
 
-
     try:
         logger.debug(f"[Gemini Fallback] Initializing model {GEMINI_MODEL}")
         model = genai.GenerativeModel(GEMINI_MODEL)
@@ -359,21 +353,12 @@ async def generate_summary_gemini(text: str, summary_type: str) -> str:
         response = await model.generate_content_async( full_prompt, safety_settings=safety_settings, request_options=request_options )
         logger.info("[Gemini Fallback] Received response from Gemini.")
 
-        # Checking block reason is still useful for debugging, though unlikely now
-        if response.prompt_feedback.block_reason: logger.warning(f"[Gemini Fallback] Request blocked unexpectedly. Reason: {response.prompt_feedback.block_reason}"); # Changed to warning
-        # Check candidates for safety blocks as well
+        if response.prompt_feedback.block_reason: logger.warning(f"[Gemini Fallback] Request blocked unexpectedly. Reason: {response.prompt_feedback.block_reason}");
         for cand in response.candidates:
-            if cand.finish_reason == 'SAFETY':
-                logger.warning(f"[Gemini Fallback] Candidate blocked due to SAFETY. Ratings: {cand.safety_ratings}")
+            if cand.finish_reason == 'SAFETY': logger.warning(f"[Gemini Fallback] Candidate blocked due to SAFETY. Ratings: {cand.safety_ratings}")
 
-
-        # Attempt to get text even if blocked (might be None)
-        try:
-            summary = response.text
-        except ValueError as e:
-            # Handle cases where accessing response.text raises ValueError (e.g., blocked content)
-            logger.warning(f"[Gemini Fallback] Error accessing response text (likely blocked content): {e}")
-            summary = None # Ensure summary is None
+        try: summary = response.text
+        except ValueError as e: logger.warning(f"[Gemini Fallback] Error accessing response text (likely blocked content): {e}"); summary = None
 
         if summary: logger.info(f"[Gemini Fallback] Success generating summary. Output len: {len(summary)}"); return summary.strip().replace('_', r'\_').replace('*', r'\*').replace('[', r'\[').replace('`', r'\`')
         else: logger.warning(f"[Gemini Fallback] Gemini returned empty summary or content was blocked. Finish reason: {response.candidates[0].finish_reason if response.candidates else 'N/A'}"); return "Sorry, the fallback AI model did not provide a summary (possibly due to content filters)."
@@ -392,7 +377,7 @@ async def generate_summary(text: str, summary_type: str) -> str:
     if len(text) > MAX_INPUT_LENGTH: logger.warning(f"[Primary Summary] Input length ({len(text)}) exceeds limit ({MAX_INPUT_LENGTH}). Truncating."); text = text[:MAX_INPUT_LENGTH] + "... (Content truncated)"
     full_prompt = f"{prompt}\n\n{text}"
     headers = { "Authorization": f"Bearer {OPENROUTER_API_KEY}", "Content-Type": "application/json" }; payload = { "model": OPENROUTER_MODEL, "messages": [{"role": "user", "content": full_prompt}] }; openrouter_api_endpoint = "https://openrouter.ai/api/v1/chat/completions"
-    api_timeouts = httpx.Timeout(15.0, read=10.0, write=15.0, pool=60.0) # 10 second read timeout
+    api_timeouts = httpx.Timeout(15.0, read=10.0, write=15.0, pool=60.0)
     response = None
 
     try:
@@ -421,7 +406,10 @@ async def generate_summary(text: str, summary_type: str) -> str:
                 elif response.status_code == 429: logger.warning("[Primary Summary] OpenRouter Rate Limit Exceeded (429)."); return "Sorry, primary AI model is busy. Try again."
                 elif response.status_code == 500: logger.error(f"[Primary Summary] OpenRouter Internal Server Error (500). Resp:{response.text[:500]}"); return "Sorry, primary AI service internal error."
                 else:
-                    error_info = ""; try: error_info = response.json().get("error", {}).get("message", "") except Exception: pass
+                    error_info = "";
+                    try: # Correct Indentation
+                        error_info = response.json().get("error", {}).get("message", "")
+                    except Exception: pass
                     logger.error(f"[Primary Summary] Unexpected status {response.status_code} from OpenRouter. Error: '{error_info}' Resp:{response.text[:500]}");
                     return f"Sorry, primary AI service returned unexpected status ({response.status_code})."