Update notifier.py

notifier.py

@@ -4,6 +4,7 @@ import requests
 from email.mime.text import MIMEText
 from email.mime.multipart import MIMEMultipart
 
+
 EMAIL_USER = os.getenv("EMAIL_USER")
 EMAIL_PASS = os.getenv("EMAIL_PASS")
 
@@ -13,13 +14,37 @@ def get_user_ip():
     except requests.RequestException:
         return "Unavailable"
 
-def send_email_notification(ip_address, user_message):
+def get_geo_from_ip(ip):
+    try:
+        response = requests.get(f"http://ip-api.com/json/{ip}", timeout=5)
+        data = response.json()
+        if data["status"] == "success":
+            return {
+                "city": data.get("city", "Unknown"),
+                "region": data.get("regionName", "Unknown"),
+                "country": data.get("country", "Unknown"),
+                "error": ""
+            }
+        else:
+            return {"error": "Lookup failed"}
+    except Exception as e:
+        return {"error": str(e)}
+
+def send_email_notification(user_message):
     if not EMAIL_USER or not EMAIL_PASS:
         print("[Notifier] Missing EMAIL_USER or EMAIL_PASS.")
         return
 
+    ip_address = _get_user_ip()
+    location = _get_geo_from_ip(ip_address)
+
+    if location["error"]:
+        location_info = f"IP: {ip_address} (location lookup failed)"
+    else:
+        location_info = f"{location['city']}, {location['region']}, {location['country']} (IP: {ip_address})"
+
     subject = "RezAI Used"
-    body = f"A new user interacted with RezAi.\n\nIP Address: {ip_address}\nMessage: {user_message}"
+    body = f"A new user interacted with RezAi.\n\nLocation: {location_info}\nMessage: {user_message}"
 
     msg = MIMEMultipart()
     msg["From"] = EMAIL_USER
@@ -34,3 +59,4 @@ def send_email_notification(ip_address, user_message):
             server.sendmail(EMAIL_USER, EMAIL_USER, msg.as_string())
     except Exception as e:
         print(f"[Notifier] Email sending error: {e}")
+