Upload index.py

index.py

@@ -5,6 +5,8 @@ import time
 import random
 import os
 
+
+
 API_URL = os.getenv("API_URL")
 
 TRANSITION_MESSAGES = [
@@ -15,41 +17,14 @@ TRANSITION_MESSAGES = [
     "Onward to question {number}!"
 ]
 
-TAB1_FILE = "tab1.txt"
-TAB1_FALLBACK = """
-# Registration System Demo
-
-Data quality issues have long plagued council databases, burdening W2 workflow teams (e.g. the Assisted Travel Team's dashboard I am working on), data analysts, and engineers with manual cleaning and validation. This pilot explores how AI-driven workflows can address these pain points by streamlining user registration, ensuring accurate data collection, and reducing validation overhead.
-
-*If you see this message, `tab1.txt` was not found. Please upload it to the app directory for the full intro page.*
-"""
-
-def read_content_from_file(file_path):
-    try:
-        import os
-        st.write("Current working directory:", os.getcwd())
-        st.write("Files in directory:", os.listdir())
-        with open(file_path, 'r') as file:
-            return file.read()
-    except FileNotFoundError:
-        return TAB1_FALLBACK
-
-def reset_registration_state():
-    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():
-    reset_registration_state()
+    print("Starting registration...")
     try:
         headers = {"Origin": "https://entz-council-3.hf.space"}
-        response = requests.post(f"{API_URL}/start_registration", headers=headers, timeout=2)
+        response = requests.post(f"{API_URL}/start_registration", headers=headers, timeout=1)
         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 = ""
@@ -59,10 +34,14 @@ 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.")
@@ -77,10 +56,12 @@ 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 = ""
@@ -97,6 +78,7 @@ def submit_response():
         st.session_state.skip_phone = False
         st.rerun()
     except requests.RequestException as e:
+        print(f"Error submitting response: {e}")
         st.error(f"Error submitting response: {e}")
 
 def edit_field(field, value):
@@ -108,10 +90,12 @@ 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":
@@ -120,57 +104,63 @@ def edit_field(field, value):
             st.success("Database updated.")
             st.rerun()
     except requests.RequestException as e:
+        print(f"Error editing field: {e}")
         st.error(f"Error editing field: {e}")
 
-def ensure_registration_state():
-    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 = {}
-
-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]**")
-
-    if st.session_state.summary:
-        st.success("Registration Complete!")
-        st.subheader("Summary")
-        for key, value in st.session_state.summary.items():
-            edit_key = f"edit_{key}"
-            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"):
-                start_registration()
-                st.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.")
-
-    elif st.session_state.current_question:
-        if st.session_state.feedback and st.session_state.feedback != "Registration complete!":
+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:
             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.strip().lower() == "what is your address?"
-        is_phone_question = st.session_state.current_question.strip().lower() == "what is your phone number?"
+        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(
@@ -179,27 +169,23 @@ def run_registration_app():
                 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="skip_address_checkbox")
+                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="skip_phone_checkbox")
+                st.session_state.skip_phone = st.checkbox("Skip this question", value=st.session_state.skip_phone)
 
-        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
-        )
+        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)
 
-        if st.button("Submit", key=f"submit_button_q{st.session_state.question_number}"):
-            submit_response()
+        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:
-        st.info("Initializing registration session...")
-
-tab1, tab2 = st.tabs(["Intro", "Registration"])
-
-with tab1:
-    st.markdown(read_content_from_file(TAB1_FILE))
-
-with tab2:
-    run_registration_app()
+        print("Waiting for session to initialize...")