ap

.env

@@ -3,4 +3,5 @@ SMTP_SERVER=smtp.gmail.com
 SMTP_PORT=587
 SMTP_USERNAME=adityaadaki21@gmail.com
 SMTP_PASSWORD=beqk eefx fmir izxz
-NOTIFICATION_EMAIL=adityaadaki21@gmail.com 
+NOTIFICATION_EMAIL=adityaadaki21@gmail.com
+TELEGRAM_CHAT_ID=5397241102 

app.py

@@ -4,10 +4,9 @@ import base64
 import os
 from dotenv import load_dotenv
 from werkzeug.utils import secure_filename
-import smtplib
-from email.mime.text import MIMEText
-from email.mime.multipart import MIMEMultipart
 import re
+import asyncio
+from telegram import Bot
 
 # Load environment variables
 load_dotenv()
@@ -21,40 +20,26 @@ app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
 model = genai.GenerativeModel("gemini-1.5-flash")
 
-# Email Configuration
-SMTP_SERVER = os.getenv('SMTP_SERVER')
-SMTP_PORT = int(os.getenv('SMTP_PORT'))
-SMTP_USERNAME = os.getenv('SMTP_USERNAME')
-SMTP_PASSWORD = os.getenv('SMTP_PASSWORD')
-NOTIFICATION_EMAIL = os.getenv('NOTIFICATION_EMAIL')
+# Telegram Configuration
+TELEGRAM_BOT_TOKEN = "7868898974:AAH2p7sBkj0TH4sD__giOmUt-2ZXn7sg0_c"
+TELEGRAM_CHAT_ID = os.getenv('TELEGRAM_CHAT_ID')  # Add this to your .env file
+bot = Bot(token=TELEGRAM_BOT_TOKEN)
 
-def send_invalid_certificate_notification(cert_name, details):
-    """Send email notification for invalid certificate."""
+async def send_invalid_certificate_notification(cert_name, details):
+    """Send Telegram notification for invalid certificate."""
     try:
-        msg = MIMEMultipart()
-        msg['From'] = SMTP_USERNAME
-        msg['To'] = NOTIFICATION_EMAIL
-        msg['Subject'] = f'Invalid Certificate Detection - {cert_name}'
-        
-        body = f"""
-        An invalid certificate has been detected:
-        
-        Certificate Name: {cert_name}
-        Details: {details}
-        
-        Please review this certificate as soon as possible.
-        """
-        
-        msg.attach(MIMEText(body, 'plain'))
-        
-        with smtplib.SMTP(SMTP_SERVER, SMTP_PORT) as server:
-            server.starttls()
-            server.login(SMTP_USERNAME, SMTP_PASSWORD)
-            server.send_message(msg)
-            
+        message = f"""
+🚨 Invalid Certificate Detected 🚨
+
+Certificate Name: {cert_name}
+Details: {details}
+
+Please review this certificate as soon as possible.
+"""
+        await bot.send_message(chat_id=TELEGRAM_CHAT_ID, text=message, parse_mode='HTML')
         return True
     except Exception as e:
-        print(f"Failed to send email notification: {str(e)}")
+        print(f"Failed to send Telegram notification: {str(e)}")
         return False
 
 def check_validity_from_response(response_text):
@@ -126,7 +111,11 @@ def process_certificate(image_path, cert_name):
     
     # Check if certificate is invalid
     if check_validity_from_response(response.text):
-        send_invalid_certificate_notification(cert_name, response.text)
+        # Create event loop and run the async notification
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        loop.run_until_complete(send_invalid_certificate_notification(cert_name, response.text))
+        loop.close()
 
     return response.text
 

requirements.txt

@@ -1,5 +1,5 @@
 flask
 python-dotenv
 google-generativeai
-aiogram>=3.0.0
-aiohttp
+Werkzeug
+python-telegram-bot