Upload index.py

index.py

@@ -3,9 +3,13 @@ import requests
 import json
 import time
 import random
+from dotenv import load_dotenv
 import os
 
-API_URL = os.getenv("API_URL")
+
+
+load_dotenv()
+API_URL = os.getenv("API_URL", "http://localhost:8000")
 
 TRANSITION_MESSAGES = [
     "Now we are moving onto question {number}!",
@@ -112,72 +116,64 @@ if "session_state" not in st.session_state:
     st.session_state.prev_question = ""
     st.session_state.question_number = 1
 
-# Create two tabs
-tab1, tab2 = st.tabs(["Introduction", "Registration"])
+if st.session_state.session_id is None:
+    start_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)
-    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
+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)
+            elif is_phone_question:
+                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"]');
+                    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"):
+            submit_response()
     else:
-        if st.session_state.current_question:
-            with st.form(key=f"question_form_{st.session_state.question_number}"):
-                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)
-                    elif is_phone_question:
-                        st.session_state.skip_phone = st.checkbox("Skip this question", value=st.session_state.skip_phone)
-
-                st.session_state.answer = st.text_input("Your Answer", value="", key=f"answer_input_{st.session_state.question_number}")
-
-                if st.form_submit_button("Submit", key="submit_button"):
-                    submit_response()
-        else:
-            print("Waiting for session to initialize...")
+        print("Waiting for session to initialize...")