Update login.py

login.py

@@ -1,29 +1,53 @@
 import streamlit as st
 import random
+import smtplib
+import os
+from email.mime.text import MIMEText
+from dotenv import load_dotenv
 from database import create_table, add_user
 
+load_dotenv()
 create_table()
 
 st.title("💪 FitPlan AI Login")
 
-email = st.text_input("Enter Email")
+EMAIL_USER = os.getenv("EMAIL_USER")
+EMAIL_PASS = os.getenv("EMAIL_PASS")
 
 if "otp" not in st.session_state:
     st.session_state.otp = None
 
-# Generate OTP
+if "logged_in" not in st.session_state:
+    st.session_state.logged_in = False
+
+email = st.text_input("Enter Email")
+
+
+def send_otp(email, otp):
+
+    msg = MIMEText(f"Your FitPlan AI OTP is: {otp}")
+    msg["Subject"] = "FitPlan AI Login OTP"
+    msg["From"] = EMAIL_USER
+    msg["To"] = email
+
+    server = smtplib.SMTP_SSL("smtp.gmail.com", 465)
+    server.login(EMAIL_USER, EMAIL_PASS)
+    server.sendmail(EMAIL_USER, email, msg.as_string())
+    server.quit()
+
+
 if st.button("Send OTP"):
 
     if email:
 
-        otp = random.randint(100000,999999)
-
+        otp = random.randint(100000, 999999)
         st.session_state.otp = otp
-        st.success(f"Your OTP is: {otp}")
 
-        # Save user in database
+        send_otp(email, otp)
         add_user(email)
 
+        st.success("OTP sent to your email")
+
     else:
         st.warning("Enter email first")
 
@@ -35,7 +59,9 @@ if st.button("Verify OTP"):
     if str(st.session_state.otp) == user_otp:
 
         st.session_state.logged_in = True
-        st.success("Login Successful 🎉")
+        st.success("Login successful 🎉")
+
+        st.switch_page("app.py")
 
     else:
         st.error("Invalid OTP")