Upload index.py

index.py

@@ -5,8 +5,7 @@ import time
 import random
 import os
 
-
-
+# --- CONFIGURATION ---
 API_URL = os.getenv("API_URL")
 
 TRANSITION_MESSAGES = [
@@ -17,14 +16,34 @@ TRANSITION_MESSAGES = [
     "Onward to question {number}!"
 ]
 
+TAB1_FILE = "tab1.txt"
+
+# --- HELPER FUNCTIONS ---
+
+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."
+
+def reset_registration_state():
+    # Remove all registration-related session state keys
+    for key in [
+        "session_id", "current_question", "answer", "feedback", "summary",
+        "skip_address", "skip_phone", "prev_question", "question_number",
+        "edit_field_values"
+    ]:
+        if key in st.session_state:
+            del st.session_state[key]
+
 def start_registration():
-    print("Starting registration...")
+    reset_registration_state()
     try:
         headers = {"Origin": "https://entz-council-3.hf.space"}
-        response = requests.post(f"{API_URL}/start_registration", headers=headers, timeout=1)
+        response = requests.post(f"{API_URL}/start_registration", headers=headers, timeout=2)
         response.raise_for_status()
         data = response.json()
-        print("API Response:", data)
         st.session_state.session_id = data["session_id"]
         st.session_state.current_question = data["message"]
         st.session_state.feedback = ""
@@ -34,14 +53,10 @@ def start_registration():
         st.session_state.skip_phone = False
         st.session_state.prev_question = ""
         st.session_state.question_number = 1
+        st.session_state.edit_field_values = {}
     except requests.RequestException as e:
-        print(f"Error starting registration: {e}, Response: {getattr(e.response, 'text', 'No response')}")
         st.error(f"Error starting registration: {e}")
-        
-        
-        
-        
-        
+
 def submit_response():
     if not st.session_state.session_id:
         st.error("No active session. Please start registration.")
@@ -56,12 +71,10 @@ def submit_response():
         "answer": st.session_state.answer,
         "skip_steps": skip_steps
     }
-    print("Submitting response with payload:", payload)
     try:
         response = requests.post(f"{API_URL}/submit_response", json=payload)
         response.raise_for_status()
         data = response.json()
-        print("API Response:", data)
         if data.get("message") == "Registration complete!":
             st.session_state.summary = data["summary"]
             st.session_state.current_question = ""
@@ -76,9 +89,8 @@ def submit_response():
             st.session_state.answer = ""
         st.session_state.skip_address = False
         st.session_state.skip_phone = False
-        st.rerun()
+        st.experimental_rerun()
     except requests.RequestException as e:
-        print(f"Error submitting response: {e}")
         st.error(f"Error submitting response: {e}")
 
 def edit_field(field, value):
@@ -90,77 +102,78 @@ def edit_field(field, value):
         "field_to_edit": field,
         "new_value": value
     }
-    print("Editing field with payload:", payload)
     try:
         response = requests.post(f"{API_URL}/edit_field", json=payload)
         response.raise_for_status()
         data = response.json()
-        print("API Response:", data)
         st.session_state.feedback = data.get("validation_feedback", "")
         st.session_state.summary = data.get("summary", st.session_state.summary)
         if data.get("message") == "Needs clarification":
             st.error(f"Clarification needed for {field}: {data['validation_feedback']}")
         elif data.get("message") == "Field updated successfully!":
             st.success("Database updated.")
-            st.rerun()
+            st.experimental_rerun()
     except requests.RequestException as e:
-        print(f"Error editing field: {e}")
         st.error(f"Error editing field: {e}")
 
-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
-
-if st.session_state.session_id is None:
-    start_registration()
-
-# 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."
-
-st.title("AI-Powered Registration System")
-st.markdown("If 403 or other connection errors, please refresh the page every 1 minute, because the backend server is being spun up. Developed by [email protected]")
-
-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:
+def ensure_registration_state():
+    # Initialize session state if not already present
+    if "session_id" not in st.session_state:
+        start_registration()
+    if "edit_field_values" not in st.session_state:
+        st.session_state.edit_field_values = {}
+
+# --- MAIN APP LOGIC ---
+
+def run_registration_app():
+    ensure_registration_state()
+
+    st.title("AI-Powered Registration System")
+    st.markdown("*** If 403 or other connection errors, please refresh the page every 1 minute, because the backend server is being spun up. Developed by [email protected]**")
+
+    # --- REGISTRATION COMPLETE ---
+    if st.session_state.summary:
+        st.success("Registration Complete!")
+        st.subheader("Summary")
+        for key, value in st.session_state.summary.items():
+            # Use a unique key for each edit field
+            edit_key = f"edit_{key}"
+            # Use session state to persist edit field values
+            if edit_key not in st.session_state.edit_field_values:
+                st.session_state.edit_field_values[edit_key] = ""
+            st.write(f"**{key}**: {value}")
+            st.session_state.edit_field_values[edit_key] = st.text_input(
+                f"Edit {key}",
+                value=st.session_state.edit_field_values[edit_key],
+                key=edit_key
+            )
+            if st.button(f"Update {key}", key=f"update_{key}"):
+                edit_field(key, st.session_state.edit_field_values[edit_key])
+
+        col1, col2 = st.columns(2)
+        with col1:
+            if st.button("Next Registration", key="next_reg"):
+                # Reset registration and rerun
+                start_registration()
+                st.experimental_rerun()
+        with col2:
+            if st.button("End Session", key="end_sess"):
+                reset_registration_state()
+                st.success("Session ended. Please select 'Registration' tab to start again.")
+
+    # --- REGISTRATION IN PROGRESS ---
+    elif st.session_state.current_question:
+        if st.session_state.feedback and st.session_state.feedback != "Registration complete!":
             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?"
+        is_address_question = st.session_state.current_question.strip().lower() == "what is your address?"
+        is_phone_question = st.session_state.current_question.strip().lower() == "what is your phone number?"
 
         if is_address_question or is_phone_question:
             st.info(
@@ -169,23 +182,30 @@ else:
                 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)
+                st.session_state.skip_address = st.checkbox("Skip this question", value=st.session_state.skip_address, key="skip_address_checkbox")
             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.skip_phone = st.checkbox("Skip this question", value=st.session_state.skip_phone, key="skip_phone_checkbox")
 
-        st.session_state.answer = st.text_input("Your Answer", value="", key=f"answer_input_{st.session_state.question_number}")
+        # Unique key for each question to avoid double entry
+        answer_key = f"answer_input_q{st.session_state.question_number}"
+        st.session_state.answer = st.text_input(
+            "Your Answer",
+            value=st.session_state.answer,
+            key=answer_key
+        )
 
-        if st.button("Submit", key="submit_button"):
+        if st.button("Submit", key=f"submit_button_q{st.session_state.question_number}"):
             submit_response()
+
     else:
-        print("Waiting for session to initialize...")
+        st.info("Initializing registration session...")
+
+# --- STREAMLIT TABS ---
+
+tab1, tab2 = st.tabs(["Intro", "Registration"])
+
+with tab1:
+    st.markdown(read_content_from_file(TAB1_FILE))
+
+with tab2:
+    run_registration_app()