Update app.py

app.py

@@ -2,7 +2,11 @@ import streamlit as st
 from transformers import pipeline
 from PyPDF2 import PdfReader
 from sentence_transformers import SentenceTransformer, util
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
 import os
+import random
 
 # Load Hugging Face model (e.g., FLAN-T5 or GPT-like model)
 @st.cache_resource
@@ -34,6 +38,37 @@ def get_relevant_content(query, texts, embedder, embeddings):
     best_idx = scores.argmax().item()
     return texts[best_idx]
 
+# Email functionality
+def send_email(to_email, subject, body):
+    try:
+        sender_email = "your-email@example.com"  # Replace with your email
+        sender_password = "your-email-password"  # Replace with your email password
+        smtp_server = "smtp.gmail.com"  # Adjust for your email provider
+        smtp_port = 587
+        
+        # Set up the email
+        msg = MIMEMultipart()
+        msg['From'] = sender_email
+        msg['To'] = to_email
+        msg['Subject'] = subject
+        msg.attach(MIMEText(body, 'plain'))
+        
+        # Send the email
+        with smtplib.SMTP(smtp_server, smtp_port) as server:
+            server.starttls()
+            server.login(sender_email, sender_password)
+            server.send_message(msg)
+        return True
+    except Exception as e:
+        st.error(f"Failed to send email: {e}")
+        return False
+
+# CAPTCHA generation
+def generate_captcha():
+    num1 = random.randint(1, 9)
+    num2 = random.randint(1, 9)
+    return num1, num2, num1 + num2
+
 # Streamlit UI
 def main():
     st.title("Educational Assistant Chatbot")
@@ -64,7 +99,7 @@ def main():
             response = model(f"Question: {user_query} Context: {relevant_content}", max_length=200)
             st.success(response[0]['generated_text'])
 
-    # Profile creation
+    # Profile creation with CAPTCHA
     st.markdown("### Create a Student Profile")
     name = st.text_input("Name:")
     email = st.text_input("Email:")
@@ -74,20 +109,31 @@ def main():
     career_goal = st.text_area("Career Goals:")
     visa_query = st.text_area("Visa Concerns or Questions:")
 
+    # CAPTCHA generation
+    if "captcha_result" not in st.session_state:
+        num1, num2, st.session_state.captcha_result = generate_captcha()
+    st.markdown(f"**CAPTCHA: {num1} + {num2} = ?**")
+    captcha_input = st.text_input("Enter CAPTCHA Result:")
+
     if st.button("Submit Profile"):
-        profile = f"""
-        Name: {name}
-        Email: {email}
-        Contact Number: {contact_number}
-        Level of Study: {study_level}
-        Field of Interest: {field_of_interest}
-        Career Goals: {career_goal}
-        Visa Queries: {visa_query}
-        """
-        # Simulate email sending (for simplicity, just display the profile)
-        st.markdown("### Profile Summary")
-        st.code(profile)
-        st.success(f"Profile submitted to {email_to_send}!")
+        if int(captcha_input) != st.session_state.captcha_result:
+            st.error("Invalid CAPTCHA. Please try again.")
+        else:
+            profile = f"""
+            Name: {name}
+            Email: {email}
+            Contact Number: {contact_number}
+            Level of Study: {study_level}
+            Field of Interest: {field_of_interest}
+            Career Goals: {career_goal}
+            Visa Queries: {visa_query}
+            """
+            # Send profile via email
+            email_sent = send_email(email_to_send, "New Student Profile Submission", profile)
+            if email_sent:
+                st.success(f"Profile submitted successfully to {email_to_send}!")
+            else:
+                st.error("Failed to submit the profile. Please try again later.")
 
 if __name__ == "__main__":
     main()