Update index.py

index.py

@@ -1,11 +1,10 @@
+import os
+os.environ["HOME"] = "/tmp"  # <--- Fix for HuggingFace/permissions
+
 import streamlit as st
 import requests
-import json
 import time
 import random
-import os
-
-
 
 API_URL = os.getenv("API_URL")
 
@@ -17,11 +16,24 @@ TRANSITION_MESSAGES = [
     "Onward to question {number}!"
 ]
 
+def reset_session_state():
+    keys = [
+        "session_id", "current_question", "answer", "feedback", "summary",
+        "skip_address", "skip_phone", "prev_question", "question_number"
+    ]
+    for key in keys:
+        if key in st.session_state:
+            del st.session_state[key]
+
 def start_registration():
     print("Starting registration...")
     try:
         headers = {"Origin": "https://entz-council-3.hf.space"}
-        response = requests.post(f"{API_URL}/start_registration", headers=headers, timeout=5)
+        response = requests.post(
+            f"{API_URL}/start_registration",
+            headers=headers,
+            timeout=10
+        )
         response.raise_for_status()
         data = response.json()
         print("API Response:", data)
@@ -37,11 +49,7 @@ def start_registration():
     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.")
@@ -58,7 +66,11 @@ def submit_response():
     }
     print("Submitting response with payload:", payload)
     try:
-        response = requests.post(f"{API_URL}/submit_response", json=payload)
+        response = requests.post(
+            f"{API_URL}/submit_response",
+            json=payload,
+            timeout=10
+        )
         response.raise_for_status()
         data = response.json()
         print("API Response:", data)
@@ -92,7 +104,11 @@ def edit_field(field, value):
     }
     print("Editing field with payload:", payload)
     try:
-        response = requests.post(f"{API_URL}/edit_field", json=payload)
+        response = requests.post(
+            f"{API_URL}/edit_field",
+            json=payload,
+            timeout=10
+        )
         response.raise_for_status()
         data = response.json()
         print("API Response:", data)
@@ -107,8 +123,7 @@ def edit_field(field, value):
         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 = {}
+if "session_id" not in st.session_state:
     st.session_state.session_id = None
     st.session_state.current_question = ""
     st.session_state.answer = ""
@@ -122,16 +137,8 @@ if "session_state" not in st.session_state:
 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]")
+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!")
@@ -143,11 +150,14 @@ if st.session_state.summary:
             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"):
+            reset_session_state()
+            start_registration()
+            st.rerun()
     with col2:
-        if st.button("End Session", key="end_sess", on_click=start_registration):
-            pass
+        if st.button("End Session", key="end_sess"):
+            reset_session_state()
+            st.success("Session ended. You may close the tab.")
 else:
     if st.session_state.current_question:
         if st.session_state.feedback:
@@ -156,7 +166,6 @@ else:
             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?"
@@ -173,19 +182,13 @@ else:
             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}")
+        st.session_state.answer = st.text_input(
+            "Your Answer",
+            value=st.session_state.answer,
+            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...")
+        st.info("Initializing session, please wait...")