Upload index.py

index.py

@@ -76,8 +76,8 @@ def submit_response():
             if st.session_state.prev_question != st.session_state.current_question:
                 st.session_state.question_number += 1
             st.session_state.answer = ""
-        st.session_state.skip_address = False
-        st.session_state.skip_phone = False
+            st.session_state.skip_address = False
+            st.session_state.skip_phone = False
         st.rerun()
     except requests.RequestException as e:
         print(f"Error submitting response: {e}")
@@ -111,23 +111,18 @@ def edit_field(field, value):
 
 def main():
     st.title("AI-Powered Registration System")
-
     # Create two tabs
     tab1, tab2 = st.tabs(["Introduction", "Registration"])
-
     # Tab 1: Introduction
     with tab1:
-        st.subheader("About This Application")
+        # st.subheader("About This Application")
         intro_content = read_content_from_file("tab1.txt")
         st.markdown(intro_content)
-
     # Tab 2: Registration Application
     with tab2:
-        # st.subheader("Registration")
         registration_content = read_content_from_file("tab2.txt")
         st.markdown(registration_content)
         st.markdown("**Developed by [email protected]**")
-
         if st.session_state.get("summary"):
             st.success("Registration Complete!")
             st.subheader("Summary")
@@ -138,11 +133,15 @@ def main():
                     edit_field(key, new_value)
             col1, col2 = st.columns(2)
             with col1:
-                if st.button("Next Registration", key="next_reg", on_click=start_registration):
-                    pass
+                if st.button("Next Registration", key="next_reg"):
+                    start_registration()
             with col2:
-                if st.button("End Session", key="end_sess", on_click=start_registration):
-                    pass
+                if st.button("End Session", key="end_sess"):
+                    # Custom logic for ending session can be added here
+                    st.session_state.session_id = None
+                    st.session_state.summary = None
+                    st.session_state.current_question = ""
+                    st.rerun()
         else:
             if st.session_state.get("current_question"):
                 if st.session_state.get("feedback"):
@@ -152,42 +151,36 @@ def main():
                     st.info(transition_msg)
                     time.sleep(1)
                 st.subheader(f"Question {st.session_state.question_number}: {st.session_state.current_question}")
-
                 is_address_question = st.session_state.current_question == "What is your address?"
                 is_phone_question = st.session_state.current_question == "What is your phone number?"
-
                 if is_address_question or is_phone_question:
                     st.info(
                         f"This question is optional. Check the box to skip, or enter your information below.\n"
                         f"- For address: Include house number (e.g., 123), street name (e.g., High Street), town/city (e.g., London), and postcode (e.g., SW1A 1AA). Example: 123 High Street, London, SW1A 1AA.\n"
                         f"- For phone: Use 10 digits for landlines (e.g., 020 123 4567) or 11 digits for mobiles starting with 07 (e.g., 07700 900 123). Avoid +44 formats."
                     )
-                    if is_address_question:
-                        st.session_state.skip_address = st.checkbox("Skip this question", value=st.session_state.skip_address)
-                    elif is_phone_question:
-                        st.session_state.skip_phone = st.checkbox("Skip this question", value=st.session_state.skip_phone)
-
+                if is_address_question:
+                    st.session_state.skip_address = st.checkbox("Skip this question", value=st.session_state.skip_address, key=f"skip_addr_{st.session_state.question_number}")
+                elif is_phone_question:
+                    st.session_state.skip_phone = st.checkbox("Skip this question", value=st.session_state.skip_phone, key=f"skip_phone_{st.session_state.question_number}")
                 st.components.v1.html("""
-                    document.addEventListener('keypress', function(e) {
-                        if (e.key === 'Enter' && document.activeElement.tagName === 'INPUT') {
-                            e.preventDefault();
-                            const submitButton = document.querySelector('button[key="submit_button"]');
-                            if (submitButton && !submitButton.disabled) submitButton.click();
-                        }
-                    });
+                document.addEventListener('keypress', function(e) {
+                    if (e.key === 'Enter' && document.activeElement.tagName === 'INPUT') {
+                        e.preventDefault();
+                        const submitButton = document.querySelector('button[key="submit_button_' + window.st.session_state.question_number + '"]');
+                        if (submitButton && !submitButton.disabled) submitButton.click();
+                    }
+                });
                 """, height=0)
-
                 st.session_state.answer = st.text_input("Your Answer", value="", key=f"answer_input_{st.session_state.question_number}")
-
-                if st.button("Submit", key="submit_button"):
+                if st.button("Submit", key=f"submit_button_{st.session_state.question_number}"):
                     submit_response()
             else:
                 print("Waiting for session to initialize...")
 
 if __name__ == "__main__":
-    # Initialize session state
-    if "session_state" not in st.session_state:
-        st.session_state.session_state = {}
+    # Initialize session state (correctly)
+    if "session_id" not in st.session_state:
         st.session_state.session_id = None
         st.session_state.current_question = ""
         st.session_state.answer = ""
@@ -197,8 +190,5 @@ if __name__ == "__main__":
         st.session_state.skip_phone = False
         st.session_state.prev_question = ""
         st.session_state.question_number = 1
-
-    if st.session_state.session_id is None:
         start_registration()
-
-    main()
+    main()