Upload index.py

index.py

@@ -5,17 +5,9 @@ import time
 import random
 import os
 
-# Load environment variables
+load_dotenv()
 API_URL = os.getenv("API_URL")
 
-# Read content from text files
-def read_content_from_file(file_path):
-    try:
-        with open(file_path, 'r') as file:
-            return file.read()
-    except FileNotFoundError:
-        return f"Error: {file_path} not found."
-
 TRANSITION_MESSAGES = [
     "Now we are moving onto question {number}!",
     "Time to switch to question {number}!",
@@ -27,7 +19,7 @@ TRANSITION_MESSAGES = [
 def start_registration():
     print("Starting registration...")
     try:
-        response = requests.post(f"{API_URL}/start_registration", timeout=1)
+        response = requests.post(f"{API_URL}/start_registration")
         response.raise_for_status()
         data = response.json()
         print("API Response:", data)
@@ -76,8 +68,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}")
@@ -109,86 +101,93 @@ def edit_field(field, value):
         print(f"Error editing field: {e}")
         st.error(f"Error editing field: {e}")
 
-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")
-        intro_content = read_content_from_file("tab1.txt")
+if "session_state" not in st.session_state:
+    st.session_state.session_state = {}
+    st.session_state.session_id = None
+    st.session_state.current_question = ""
+    st.session_state.answer = ""
+    st.session_state.feedback = ""
+    st.session_state.summary = None
+    st.session_state.skip_address = False
+    st.session_state.skip_phone = False
+    st.session_state.prev_question = ""
+    st.session_state.question_number = 1
+
+# Create two tabs
+tab1, tab2 = st.tabs(["Introduction", "Registration"])
+
+# Tab 1: Introduction
+with tab1:
+    st.header("Welcome to the AI-Powered Registration System")
+    try:
+        with open("tab1.txt", "r") as file:
+            intro_content = file.read()
         st.markdown(intro_content)
-    # Tab 2: Registration Application
-    with tab2:
-        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")
-            for key, value in st.session_state.summary.items():
-                st.write(f"**{key}**: {value}")
-                new_value = st.text_input(f"Edit {key}", key=f"edit_{key}")
-                if st.button(f"Update {key}", key=f"update_{key}"):
-                    edit_field(key, new_value)
-            col1, col2 = st.columns(2)
-            with col1:
-                if st.button("Next Registration", key="next_reg"):
-                    start_registration()
-            with col2:
-                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"):
-                    st.error(st.session_state.feedback)
-                if st.session_state.get("prev_question") and st.session_state.prev_question != st.session_state.current_question:
-                    transition_msg = random.choice(TRANSITION_MESSAGES).format(number=st.session_state.question_number)
-                    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."
-                    )
+    except FileNotFoundError:
+        st.error("Introduction file (tab1.txt) not found. Please create the file with the introduction content.")
+    except Exception as e:
+        st.error(f"Error reading introduction file: {e}")
+
+# Tab 2: Registration App
+with tab2:
+    if st.session_state.session_id is None:
+        start_registration()
+
+    st.title("AI-Powered Registration System")
+
+    if st.session_state.summary:
+        st.success("Registration Complete!")
+        st.subheader("Summary")
+        for key, value in st.session_state.summary.items():
+            st.write(f"**{key}**: {value}")
+            new_value = st.text_input(f"Edit {key}", key=f"edit_{key}")
+            if st.button(f"Update {key}", key=f"update_{key}"):
+                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
+        with col2:
+            if st.button("End Session", key="end_sess", on_click=start_registration):
+                pass
+    else:
+        if st.session_state.current_question:
+            if st.session_state.feedback:
+                st.error(st.session_state.feedback)
+            if st.session_state.prev_question and st.session_state.prev_question != st.session_state.current_question:
+                transition_msg = random.choice(TRANSITION_MESSAGES).format(number=st.session_state.question_number)
+                st.info(transition_msg)
+                time.sleep(1)
+            print(f"Displaying question: {st.session_state.current_question}")
+            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). Do not use +44 or other region numbers."
+                )
                 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}")
+                    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, key=f"skip_phone_{st.session_state.question_number}")
-                st.components.v1.html("""
+                    st.session_state.skip_phone = st.checkbox("Skip this question", value=st.session_state.skip_phone)
+
+            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_' + window.st.session_state.question_number + '"]');
+                        const submitButton = document.querySelector('button[key="submit_button"]');
                         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=f"submit_button_{st.session_state.question_number}"):
-                    submit_response()
-            else:
-                print("Waiting for session to initialize...")
-
-if __name__ == "__main__":
-    # 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 = ""
-        st.session_state.feedback = ""
-        st.session_state.summary = None
-        st.session_state.skip_address = False
-        st.session_state.skip_phone = False
-        st.session_state.prev_question = ""
-        st.session_state.question_number = 1
-        start_registration()
-    main()
+            """, 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"):
+                submit_response()
+        else:
+            print("Waiting for session to initialize...")