Update app.py

app.py

@@ -72,68 +72,72 @@ def main():
     st.title("Educational Assistant Chatbot")
     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 dataset
-    st.sidebar.markdown("### Data Loading")
-    if st.sidebar.button("Load Dataset"):
-        st.session_state.pdf_texts = extract_text_from_pdfs("chatbot")
+    # Hardcoded dataset folder and email
+    dataset_folder = "./pdf_dataset"
+    email_to_send = "application@aspireec.pk"
+
+    # Load dataset on startup
+    if "pdf_texts" not in st.session_state:
+        st.session_state.pdf_texts = extract_text_from_pdfs(dataset_folder)
         st.session_state.embedder, st.session_state.embeddings = create_embeddings(st.session_state.pdf_texts)
-        st.sidebar.success("Dataset Loaded Successfully!")
-
-    # Collect user information
-    st.markdown("### Create a Student Profile")
-    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
-    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 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"""
-            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}
-            """
-            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}!")
+
+    # Initialize session state for profile submission
+    if "profile_submitted" not in st.session_state:
+        st.session_state.profile_submitted = False
+
+    # Show profile creation form only if profile is not submitted
+    if not st.session_state.profile_submitted:
+        st.markdown("### Create a Student Profile")
+        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
+        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 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"""
+                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}
+                """
+                email_sent = send_email(email_to_send, "New Student Profile Submission", profile)
+                if email_sent:
+                    st.session_state.profile_submitted = True
+                    st.success(f"Profile submitted successfully to {email_to_send}!")
+                else:
+                    st.error("Failed to submit the profile. Please try again later.")
+
+    # Show question input only if profile is submitted
+    if st.session_state.profile_submitted:
+        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 restart the application.")
             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'])
+                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()