Update app.py

app.py

@@ -5,27 +5,16 @@ import base64
 from PIL import Image
 import io
 import datetime
-import smtplib
-from email.mime.text import MIMEText
-from email.mime.multipart import MIMEMultipart
 
 # Configuration from environment variables
 OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
 DEFAULT_REPORT_EMAIL = os.getenv("REPORT_EMAIL")
-
-# Email configuration - SMTP
-BREVO_SMTP_SERVER = os.getenv("BREVO_SMTP_SERVER")
-BREVO_SMTP_PORT = int(os.getenv("BREVO_SMTP_PORT", "587"))
-BREVO_SMTP_USERNAME = os.getenv("BREVO_SMTP_USERNAME")
-BREVO_SMTP_PASSWORD = os.getenv("BREVO_SMTP_PASSWORD")
-BREVO_FROM_EMAIL = os.getenv("BREVO_FROM_EMAIL", "noreply@yourdomain.com")
-
-# Email configuration - API (fallback)
 BREVO_API_KEY = os.getenv("BREVO_API_KEY")
+BREVO_FROM_EMAIL = os.getenv("BREVO_FROM_EMAIL", "noreply@yourdomain.com")
 
 # Testing/Demo Mode Configuration
-# Disable demo mode if SMTP or API is configured
-DEMO_MODE = os.getenv("DEMO_MODE", "false").lower() == "true" if (BREVO_SMTP_SERVER or BREVO_API_KEY) else True
+# Auto-disable demo mode if API key is configured
+DEMO_MODE = os.getenv("DEMO_MODE", "false").lower() == "true" if BREVO_API_KEY else True
 
 # Store conversation history and settings
 conversation_history = []
@@ -59,38 +48,12 @@ def update_report_email(new_email):
     else:
         return "❌ Please enter a valid email address"
 
-def send_email_smtp(to_email, subject, html_content, from_name="AI Coordination System"):
-    """Send email using SMTP"""
-    try:
-        if not all([BREVO_SMTP_SERVER, BREVO_SMTP_USERNAME, BREVO_SMTP_PASSWORD]):
-            return "❌ SMTP not configured - missing server, username, or password"
-        
-        # Create message
-        msg = MIMEMultipart('alternative')
-        msg['Subject'] = subject
-        msg['From'] = f"{from_name} <{BREVO_FROM_EMAIL}>"
-        msg['To'] = to_email
-        
-        # Attach HTML content
-        html_part = MIMEText(html_content, 'html')
-        msg.attach(html_part)
-        
-        # Send email
-        with smtplib.SMTP(BREVO_SMTP_SERVER, BREVO_SMTP_PORT) as server:
-            server.starttls()
-            server.login(BREVO_SMTP_USERNAME, BREVO_SMTP_PASSWORD)
-            server.send_message(msg)
-        
-        return f"✅ Email sent to {to_email} (via SMTP)"
-        
-    except Exception as e:
-        return f"❌ SMTP error: {str(e)}"
-
-def send_email_brevo_api(to_email, subject, html_content, from_name="AI Coordination System"):
+def send_email(to_email, subject, html_content, from_name="AI Coordination System"):
     """Send email using Brevo API"""
     try:
-        if not BREVO_API_KEY:
-            return "❌ Brevo API not configured"
+        # Demo mode: simulate email sending
+        if DEMO_MODE or not BREVO_API_KEY:
+            return f"🧪 [DEMO MODE] Email simulated to {to_email}"
         
         response = requests.post(
             "https://api.brevo.com/v3/smtp/email",
@@ -112,33 +75,13 @@ def send_email_brevo_api(to_email, subject, html_content, from_name="AI Coordina
         )
         
         if response.status_code == 201:
-            return f"✅ Email sent to {to_email} (via API)"
+            return f"✅ Email sent to {to_email}"
         else:
             error_msg = response.json().get('message', 'Unknown error')
             return f"❌ Failed to send email: {error_msg}"
             
     except Exception as e:
-        return f"❌ Email API error: {str(e)}"
-
-def send_email(to_email, subject, html_content, from_name="AI Coordination System"):
-    """Send email using available method (SMTP or API)"""
-    # Demo mode: simulate email sending
-    if DEMO_MODE:
-        return f"🧪 [DEMO MODE] Email simulated to {to_email}"
-    
-    # Try SMTP first (preferred)
-    if BREVO_SMTP_SERVER:
-        result = send_email_smtp(to_email, subject, html_content, from_name)
-        # If SMTP fails, try API as fallback
-        if "❌" in result and BREVO_API_KEY:
-            return send_email_brevo_api(to_email, subject, html_content, from_name)
-        return result
-    
-    # Fall back to API if SMTP not configured
-    if BREVO_API_KEY:
-        return send_email_brevo_api(to_email, subject, html_content, from_name)
-    
-    return "❌ No email service configured (need SMTP or API credentials)"
+        return f"❌ Email error: {str(e)}"
 
 def get_mock_ai_response(user_message, department, user_name, has_image=False):
     """Generate a mock AI response for demo/testing mode"""
@@ -391,17 +334,19 @@ Monitoring or escalation steps
                     ],
                     "temperature": 0.3
                 },
-                timeout=60
+                timeout=30  # Reduced from 60 seconds
             )
             
             if response.status_code == 200:
                 data = response.json()
                 return data["choices"][0]["message"]["content"]
             else:
-                return "I'm having trouble analyzing the image right now. Please try again in a moment."
+                # Fallback to mock response if API fails
+                return get_mock_ai_response(user_message, department, user_name, has_image=True)
                 
         except Exception as e:
-            return f"Sorry, I couldn't process the image. Error: {str(e)}"
+            # Fallback to mock response on error
+            return get_mock_ai_response(user_message, department, user_name, has_image=True)
     
     else:
         # Text-only analysis
@@ -464,17 +409,19 @@ Required monitoring or reporting actions
                     "messages": [{"role": "user", "content": prompt}],
                     "temperature": 0.3
                 },
-                timeout=30
+                timeout=20  # Reduced from 30 seconds
             )
             
             if response.status_code == 200:
                 data = response.json()
                 return data["choices"][0]["message"]["content"]
             else:
-                return "I'm having trouble analyzing your request right now. Please try again in a moment."
+                # Fallback to mock response if API fails
+                return get_mock_ai_response(user_message, department, user_name, has_image=False)
                 
         except Exception as e:
-            return f"Sorry, I couldn't process your request. Error: {str(e)}"
+            # Fallback to mock response on error
+            return get_mock_ai_response(user_message, department, user_name, has_image=False)
 
 def send_department_email(sender_dept, recipient_dept, subject, message, user_name, urgency="Normal", image=None):
     """Send email between departments with optional image attachment"""
@@ -542,15 +489,12 @@ def send_department_email(sender_dept, recipient_dept, subject, message, user_na
 def send_email_report(conversation_data):
     """Send comprehensive coordination report to administration"""
     # Demo mode check
-    if DEMO_MODE:
+    if DEMO_MODE or not BREVO_API_KEY:
         return "🧪 [DEMO MODE] Report email simulated successfully"
     
     if not current_report_email:
         return "❌ Report email not configured - please set report email address"
     
-    if not (BREVO_SMTP_SERVER or BREVO_API_KEY):
-        return "❌ Email service not configured - need SMTP or API credentials"
-    
     try:
         # Calculate department activity
         dept_activity = {}
@@ -829,15 +773,7 @@ def get_system_status():
     status_lines.append("**System Configuration:**")
     status_lines.append(f"• Demo Mode: {'🧪 ENABLED' if DEMO_MODE else '✅ DISABLED'}")
     status_lines.append(f"• OpenRouter API: {'✅ Configured' if OPENROUTER_API_KEY else '❌ Not configured (using demo)'}")
-    
-    # Email configuration status
-    if BREVO_SMTP_SERVER:
-        status_lines.append(f"• Email (SMTP): ✅ Configured ({BREVO_SMTP_SERVER}:{BREVO_SMTP_PORT})")
-    elif BREVO_API_KEY:
-        status_lines.append(f"• Email (API): ✅ Configured")
-    else:
-        status_lines.append(f"• Email: ❌ Not configured (simulated)")
-    
+    status_lines.append(f"• Brevo Email API: {'✅ Configured' if BREVO_API_KEY else '❌ Not configured (simulated)'}")
     status_lines.append(f"• Report Email: {current_report_email if current_report_email else '❌ Not set'}")
     return "\n".join(status_lines)
 
@@ -1063,4 +999,4 @@ with gr.Blocks(
     )
 
 if __name__ == "__main__":
-    ui.launch(share=True)
+    ui.launch(share=True)