Update app.py

app.py

@@ -8,7 +8,7 @@ from email.mime.multipart import MIMEMultipart
 import os
 import random
 
-# Load Hugging Face model (e.g., FLAN-T5 or GPT-like model)
+# Load Hugging Face model (e.g., FLAN-T5)
 @st.cache_resource
 def load_model():
     return pipeline("text2text-generation", model="google/flan-t5-base")
@@ -43,17 +43,15 @@ 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_server = "smtp.gmail.com"
         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)
@@ -72,51 +70,40 @@ def generate_captcha():
 # Streamlit UI
 def main():
     st.title("Educational Assistant Chatbot")
-    st.markdown("Welcome! Ask me about academic programs, visa requirements, and more!")
+    st.markdown("Welcome! Let's create your student profile to provide tailored guidance.")
 
     # Sidebar for Admin Options
     st.sidebar.header("Admin Options")
     dataset_folder = st.sidebar.text_input("Dataset Folder Path", "./pdf_dataset")
     email_to_send = st.sidebar.text_input("Email for Profile Submission", "application@aspireec.pk")
 
-    # Load model and dataset
+    # Load dataset
     st.sidebar.markdown("### Data Loading")
     if st.sidebar.button("Load Dataset"):
         st.session_state.pdf_texts = extract_text_from_pdfs("chatbot")
         st.session_state.embedder, st.session_state.embeddings = create_embeddings(st.session_state.pdf_texts)
         st.sidebar.success("Dataset Loaded Successfully!")
 
-    # Chat interface
-    user_query = st.text_input("Your Query:")
-    if st.button("Ask"):
-        if "embedder" not in st.session_state or "pdf_texts" not in st.session_state:
-            st.error("Dataset is not loaded. Please load the dataset in the sidebar.")
-        else:
-            relevant_content = get_relevant_content(
-                user_query, st.session_state.pdf_texts, st.session_state.embedder, st.session_state.embeddings
-            )
-            model = load_model()
-            response = model(f"Question: {user_query} Context: {relevant_content}", max_length=200)
-            st.success(response[0]['generated_text'])
-
-    # Profile creation with CAPTCHA
+    # Collect user information
     st.markdown("### Create a Student Profile")
-    name = st.text_input("Name:")
-    email = st.text_input("Email:")
-    contact_number = st.text_input("Contact Number (Optional):")
-    study_level = st.selectbox("Level of Study", ["Undergraduate", "Postgraduate", "PhD"])
+    name = st.text_input("Name (required):")
+    email = st.text_input("Email (required):")
+    contact_number = st.text_input("Contact Number (optional):")
+    study_level = st.selectbox("Level of Study:", ["Undergraduate", "Postgraduate", "PhD"])
     field_of_interest = st.text_input("Field of Interest:")
     career_goal = st.text_area("Career Goals:")
     visa_query = st.text_area("Visa Concerns or Questions:")
 
-    # CAPTCHA generation
+    # CAPTCHA
     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"):
-        if int(captcha_input) != st.session_state.captcha_result:
+        if not name or not email:
+            st.error("Name and Email are required fields.")
+        elif int(captcha_input) != st.session_state.captcha_result:
             st.error("Invalid CAPTCHA. Please try again.")
         else:
             profile = f"""
@@ -128,12 +115,25 @@ def main():
             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.")
 
+    # Chat interface
+    st.markdown("### Ask a Question")
+    user_query = st.text_input("Your Query:")
+    if st.button("Ask"):
+        if "embedder" not in st.session_state or "pdf_texts" not in st.session_state:
+            st.error("Dataset is not loaded. Please load the dataset in the sidebar.")
+        else:
+            relevant_content = get_relevant_content(
+                user_query, st.session_state.pdf_texts, st.session_state.embedder, st.session_state.embeddings
+            )
+            model = load_model()
+            response = model(f"Question: {user_query} Context: {relevant_content}", max_length=200)
+            st.success(response[0]['generated_text'])
+
 if __name__ == "__main__":
     main()