Update main.py

main.py

@@ -1,4 +1,4 @@
-# main.py (Correcting SyntaxErrors at lines 433-434)
+# main.py (Fixing get_secret for optional secrets)
 import os
 import re
 import logging
@@ -61,10 +61,17 @@ ptb_app: Optional[Application] = None
 # --- Environment Variable Loading & Configuration ---
 logger.info("Attempting to load secrets and configuration...")
 def get_secret(secret_name):
+    """Gets a secret from environment variables and logs status safely."""
     value = os.environ.get(secret_name)
-    log_length = min(len(value), 8) if value else 0
-    status = "Found" if value else "Not Found"
-    logger.info(f"Secret '{secret_name}': {status} (Value starts with: {value[:log_length]}...)")
+    if value:
+        status = "Found"
+        # Safely log the beginning of the value
+        log_length = min(len(value), 8)
+        value_start = value[:log_length]
+        logger.info(f"Secret '{secret_name}': {status} (Value starts with: {value_start}...)")
+    else:
+        status = "Not Found"
+        logger.warning(f"Secret '{secret_name}': {status}") # Changed to warning for not found
     return value
 
 TELEGRAM_TOKEN = get_secret('TELEGRAM_TOKEN')
@@ -72,17 +79,20 @@ OPENROUTER_API_KEY = get_secret('OPENROUTER_API_KEY')
 URLTOTEXT_API_KEY = get_secret('URLTOTEXT_API_KEY')
 SUPADATA_API_KEY = get_secret('SUPADATA_API_KEY')
 APIFY_API_TOKEN = get_secret('APIFY_API_TOKEN')
-WEBHOOK_SECRET = get_secret('WEBHOOK_SECRET')
+WEBHOOK_SECRET = get_secret('WEBHOOK_SECRET') # This will now correctly return None if not set, without erroring
 
 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")
 
+# Critical checks remain the same
 if not TELEGRAM_TOKEN: logger.critical("❌ FATAL: TELEGRAM_TOKEN not found."); raise RuntimeError("Exiting: Telegram token missing.")
 if not OPENROUTER_API_KEY: logger.error("❌ ERROR: OPENROUTER_API_KEY not found. Summarization will fail.")
 
-if not URLTOTEXT_API_KEY: logger.warning("⚠️ WARNING: URLTOTEXT_API_KEY not found.")
-if not SUPADATA_API_KEY: logger.warning("⚠️ WARNING: SUPADATA_API_KEY not found.")
-if not APIFY_API_TOKEN: logger.warning("⚠️ WARNING: APIFY_API_TOKEN not found.")
+# Optional checks remain the same (logging handled by get_secret)
+if not URLTOTEXT_API_KEY: pass # Logged as warning by get_secret
+if not SUPADATA_API_KEY: pass  # Logged as warning by get_secret
+if not APIFY_API_TOKEN: pass   # Logged as warning by get_secret
+if not WEBHOOK_SECRET: logger.info("Optional secret 'WEBHOOK_SECRET' not found. Webhook security disabled.") # Explicit info log
 
 logger.info("Secret loading and configuration check finished.")
 logger.info(f"Using OpenRouter Model: {OPENROUTER_MODEL}")
@@ -426,23 +436,16 @@ async def handle_summary_type_callback(update: Update, context: ContextTypes.DEF
     context.user_data.pop('url_to_summarize', None); context.user_data.pop('original_message_id', None); logger.debug(f"Cleared URL context for user {user.id}")
 
     global TELEGRAM_TOKEN, OPENROUTER_API_KEY
-    # --- FIX: Correct Syntax for key checks ---
     if not TELEGRAM_TOKEN:
         logger.critical("TG TOKEN missing!")
-        try:
-            await query.edit_message_text(text="❌ Bot config error.")
-        except Exception:
-            pass
-        return # Stop processing
-
+        try: await query.edit_message_text(text="❌ Bot config error.")
+        except Exception: pass
+        return
     if not OPENROUTER_API_KEY:
         logger.error("OpenRouter key missing!")
-        try:
-            await query.edit_message_text(text="❌ AI config error.")
-        except Exception:
-            pass
-        return # Stop processing
-    # --- End FIX ---
+        try: await query.edit_message_text(text="❌ AI config error.")
+        except Exception: pass
+        return
 
     logger.info(f"Scheduling task for user {user.id}, chat {query.message.chat_id}, msg {message_id_to_edit}")
     asyncio.create_task( process_summary_task( user_id=user.id, chat_id=query.message.chat_id, message_id_to_edit=message_id_to_edit, url=url, summary_type=summary_type, bot_token=TELEGRAM_TOKEN ), name=f"SummaryTask-{user.id}-{message_id_to_edit}" )