Spaces:

Legal-Assistant
/

LawBot

Running

App Files Files Community

Vishwanath77 commited on Feb 1

Commit

fcbfb48

verified ·

1 Parent(s): c857d7d

Update src/apps/email_utils.py

Browse files

Files changed (1) hide show

src/apps/email_utils.py +66 -65

src/apps/email_utils.py CHANGED Viewed

@@ -1,65 +1,66 @@
-from fastapi_mail import FastMail, MessageSchema, ConnectionConfig, MessageType
-from pydantic import EmailStr
-import os
-from dotenv import load_dotenv
-load_dotenv()
-BASE_URL = os.getenv("BASE_URL", "http://localhost:8000")
-# Support both SMTP_* and MAIL_* environment variable naming conventions
-MAIL_USERNAME = os.getenv("MAIL_USERNAME") or os.getenv("SMTP_USERNAME", "")
-MAIL_PASSWORD = os.getenv("MAIL_PASSWORD") or os.getenv("SMTP_PASSWORD", "")
-MAIL_FROM = os.getenv("MAIL_FROM") or os.getenv("FROM_EMAIL", "")
-MAIL_PORT = int(os.getenv("MAIL_PORT") or os.getenv("SMTP_PORT", "587"))
-MAIL_SERVER = os.getenv("MAIL_SERVER") or os.getenv("SMTP_SERVER", "smtp.gmail.com")
-conf = ConnectionConfig(
-    MAIL_USERNAME=MAIL_USERNAME,
-    MAIL_PASSWORD=MAIL_PASSWORD,
-    MAIL_FROM=MAIL_FROM,
-    MAIL_PORT=MAIL_PORT,
-    MAIL_SERVER=MAIL_SERVER,
-    MAIL_STARTTLS=True,
-    MAIL_SSL_TLS=False,
-    USE_CREDENTIALS=True,
-    VALIDATE_CERTS=True
-)
-async def send_reset_email(email: EmailStr, token: str):
-    # Validate email configuration before attempting to send
-    if not MAIL_USERNAME or not MAIL_PASSWORD or not MAIL_FROM:
-        missing_vars = []
-        if not MAIL_USERNAME:
-            missing_vars.append("MAIL_USERNAME or SMTP_USERNAME")
-        if not MAIL_PASSWORD:
-            missing_vars.append("MAIL_PASSWORD or SMTP_PASSWORD")
-        if not MAIL_FROM:
-            missing_vars.append("MAIL_FROM or FROM_EMAIL")
-        raise ValueError(f"Email configuration missing: {', '.join(missing_vars)}. Please set these environment variables.")
-    reset_link = f"{BASE_URL}/reset-password?token={token}"
-    html_body = f"""
-    <html>
-        <body style="font-family: Arial, sans-serif; padding: 20px; background-color: #f4f4f4;">
-            <div style="max-width: 600px; margin: 0 auto; background-color: white; padding: 30px; border-radius: 10px;">
-                <h2 style="color: #9b87f5;">Password Reset Request</h2>
-                <p>Hello,</p>
-                <p>We received a request to reset your password for your Law Bot account.</p>
-                <p>Click the button below to reset your password:</p>
-                <a href="{reset_link}" style="display: inline-block; padding: 12px 24px; background-color: #9b87f5; color: white; text-decoration: none; border-radius: 6px; margin: 20px 0;">Reset Password</a>
-                <p>Or copy and paste this link into your browser:</p>
-                <p style="color: #666; word-break: break-all;">{reset_link}</p>
-                <p style="color: #999; font-size: 12px; margin-top: 30px;">If you didn't request this, you can safely ignore this email.</p>
-            </div>
-        </body>
-    </html>
-    """
-    message = MessageSchema(
-        subject="Password Reset Request - Law Bot",
-        recipients=[email],
-        body=html_body,
-        subtype=MessageType.html
-    )
-    fm = FastMail(conf)
-    await fm.send_message(message)

+from fastapi_mail import FastMail, MessageSchema, ConnectionConfig, MessageType
+from pydantic import EmailStr
+import os
+from dotenv import load_dotenv
+load_dotenv()
+BASE_URL = os.getenv("APP_URL", "http://localhost:8000")
+# Support both SMTP_* and MAIL_* environment variable naming conventions
+MAIL_USERNAME = os.getenv("MAIL_USERNAME") or os.getenv("SMTP_USERNAME", "")
+MAIL_PASSWORD = os.getenv("MAIL_PASSWORD") or os.getenv("SMTP_PASSWORD", "")
+MAIL_FROM = os.getenv("MAIL_FROM") or os.getenv("FROM_EMAIL", "")
+MAIL_PORT = int(os.getenv("MAIL_PORT") or os.getenv("SMTP_PORT", "587"))
+MAIL_SERVER = os.getenv("MAIL_SERVER") or os.getenv("SMTP_SERVER", "smtp.gmail.com")
+conf = ConnectionConfig(
+    MAIL_USERNAME=MAIL_USERNAME,
+    MAIL_PASSWORD=MAIL_PASSWORD,
+    MAIL_FROM=MAIL_FROM,
+    MAIL_PORT=MAIL_PORT,
+    MAIL_SERVER=MAIL_SERVER,
+    MAIL_STARTTLS=True,
+    MAIL_SSL_TLS=False,
+    USE_CREDENTIALS=True,
+    VALIDATE_CERTS=True
+)
+async def send_reset_email(email: EmailStr, token: str):
+    # Validate email configuration before attempting to send
+    if not MAIL_USERNAME or not MAIL_PASSWORD or not MAIL_FROM:
+        missing_vars = []
+        if not MAIL_USERNAME:
+            missing_vars.append("MAIL_USERNAME or SMTP_USERNAME")
+        if not MAIL_PASSWORD:
+            missing_vars.append("MAIL_PASSWORD or SMTP_PASSWORD")
+        if not MAIL_FROM:
+            missing_vars.append("MAIL_FROM or FROM_EMAIL")
+        raise ValueError(f"Email configuration missing: {', '.join(missing_vars)}. Please set these environment variables.")
+    reset_link = f"{BASE_URL}/reset-password?token={token}"
+    html_body = f"""
+    <html>
+        <body style="font-family: Arial, sans-serif; padding: 20px; background-color: #f4f4f4;">
+            <div style="max-width: 600px; margin: 0 auto; background-color: white; padding: 30px; border-radius: 10px;">
+                <h2 style="color: #9b87f5;">Password Reset Request</h2>
+                <p>Hello,</p>
+                <p>We received a request to reset your password for your Law Bot account.</p>
+                <p>Click the button below to reset your password:</p>
+                <a href="{reset_link}" style="display: inline-block; padding: 12px 24px; background-color: #9b87f5; color: white; text-decoration: none; border-radius: 6px; margin: 20px 0;">Reset Password</a>
+                <p>Or copy and paste this link into your browser:</p>
+                <p style="color: #666; word-break: break-all;">{reset_link}</p>
+                <p style="color: #999; font-size: 12px; margin-top: 30px;">If you didn't request this, you can safely ignore this email.</p>
+            </div>
+        </body>
+    </html>
+    """
+    message = MessageSchema(
+        subject="Password Reset Request - Law Bot",
+        recipients=[email],
+        body=html_body,
+        subtype=MessageType.html
+    )
+    fm = FastMail(conf)
+    await fm.send_message(message)