ap123

app.py

@@ -8,6 +8,7 @@ import re
 import asyncio
 from telegram import Bot
 from telegram.error import TelegramError
+import httpx
 
 # Load environment variables
 load_dotenv()
@@ -23,15 +24,20 @@ model = genai.GenerativeModel("gemini-1.5-flash")
 
 # Telegram Configuration
 TELEGRAM_BOT_TOKEN = "7868898974:AAH2p7sBkj0TH4sD__giOmUt-2ZXn7sg0_c"
-TELEGRAM_CHAT_ID = "5397241102"  # Hardcoded chat ID
-bot = Bot(token=TELEGRAM_BOT_TOKEN)
+TELEGRAM_CHAT_ID = "5397241102"
+
+# Initialize bot with custom client
+async def init_bot():
+    async with httpx.AsyncClient(timeout=30.0) as client:
+        bot = Bot(token=TELEGRAM_BOT_TOKEN, request=client)
+        return bot
 
 async def verify_bot_connection():
     try:
+        bot = await init_bot()
         bot_info = await bot.get_me()
         print(f"✅ Successfully connected to Telegram bot: @{bot_info.username}")
         
-        # Verify chat ID
         try:
             await bot.send_message(
                 chat_id=TELEGRAM_CHAT_ID,
@@ -44,21 +50,15 @@ async def verify_bot_connection():
             print(f"Error: {chat_error}")
             return False
             
-    except TelegramError as e:
+    except Exception as e:
         print(f"❌ Failed to connect to Telegram bot")
         print(f"Error: {e}")
         return False
 
-# Initialize bot connection when the app starts
-print("🔄 Initializing Telegram bot connection...")
-loop = asyncio.new_event_loop()
-asyncio.set_event_loop(loop)
-loop.run_until_complete(verify_bot_connection())
-loop.close()
-
 async def send_invalid_certificate_notification(cert_name, details):
     """Send Telegram notification for invalid certificate."""
     try:
+        bot = await init_bot()
         message = f"""
 🚨 Invalid Certificate Detected 🚨
 
@@ -74,11 +74,21 @@ Please review this certificate as soon as possible.
         )
         print(f"✅ Successfully sent notification to Telegram")
         return True
-    except TelegramError as e:
+    except Exception as e:
         print(f"❌ Failed to send Telegram notification: {str(e)}")
         print(f"Chat ID used: {TELEGRAM_CHAT_ID}")
         return False
 
+# Initialize bot connection when the app starts
+try:
+    print("🔄 Initializing Telegram bot connection...")
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    loop.run_until_complete(verify_bot_connection())
+    loop.close()
+except Exception as e:
+    print(f"❌ Failed to initialize Telegram bot: {str(e)}")
+
 def check_validity_from_response(response_text):
     """Extract validity status from Gemini's response."""
     # Look for validity indicators in the response

requirements.txt

@@ -2,4 +2,5 @@ flask
 python-dotenv
 google-generativeai
 Werkzeug
-python-telegram-bot
+python-telegram-bot
+httpx>=0.24.0