5

.env

@@ -1 +1,6 @@
-GEMINI_API_KEY=AIzaSyCtEVix5c3pv3wtt7JLUA0CWpmOCt2gBMw
+GEMINI_API_KEY=AIzaSyCtEVix5c3pv3wtt7JLUA0CWpmOCt2gBMw
+SMTP_SERVER=smtp.gmail.com
+SMTP_PORT=587
+SMTP_USERNAME=your-email@gmail.com
+SMTP_PASSWORD=beqk eefx fmir izxz
+NOTIFICATION_EMAIL=adityaadaki21@gmail.com 

app.py

@@ -4,6 +4,13 @@ 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
+
+# Load environment variables
+load_dotenv()
 
 # Initialize Flask app
 UPLOAD_FOLDER = '/tmp/uploads'
@@ -14,6 +21,48 @@ 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')
+
+def send_invalid_certificate_notification(cert_name, details):
+    """Send email 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)
+            
+        return True
+    except Exception as e:
+        print(f"Failed to send email notification: {str(e)}")
+        return False
+
+def check_validity_from_response(response_text):
+    """Extract validity status from Gemini's response."""
+    # Look for validity indicators in the response
+    invalid_pattern = r'Invalid❌|invalid-badge'
+    return bool(re.search(invalid_pattern, response_text))
+
 def process_certificate(image_path, cert_name):
     """Process a single certificate and generate a response in structured HTML."""
     with open(image_path, "rb") as image_file:
@@ -74,6 +123,10 @@ def process_certificate(image_path, cert_name):
     response = model.generate_content([
         {'mime_type': 'image/jpeg', 'data': encoded_image}, prompt
     ])
+    
+    # Check if certificate is invalid
+    if check_validity_from_response(response.text):
+        send_invalid_certificate_notification(cert_name, response.text)
 
     return response.text