Upload index.py

index.py

@@ -26,43 +26,23 @@ TRANSITION_MESSAGES = [
 
 def start_registration():
     print("Starting registration...")
-    if "backend_waiting" not in st.session_state:
-        st.session_state.backend_waiting = False
-    if "session_id" not in st.session_state or st.session_state.session_id is None:
-        if not st.session_state.backend_waiting:
-            st.session_state.backend_waiting = True
-            st.warning("Waiting up to 60 seconds for the backend to spin up. Please wait...")
-            try:
-                response = requests.post(f"{API_URL}/start_registration", timeout=60)
-                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 = ""
-                st.session_state.summary = None
-                st.session_state.answer = ""
-                st.session_state.skip_address = False
-                st.session_state.skip_phone = False
-                st.session_state.prev_question = ""
-                st.session_state.question_number = 1
-                st.session_state.backend_waiting = False
-                st.experimental_rerun()  # Refresh to clear the warning
-            except requests.exceptions.HTTPError as e:
-                if e.response.status_code == 403:
-                    st.session_state.backend_waiting = False
-                    st.error("Backend unavailable, please try again later.")
-                else:
-                    st.session_state.backend_waiting = False
-                    st.error(f"Unexpected error: {e}")
-            except requests.RequestException as e:
-                st.session_state.backend_waiting = False
-                st.error("Connection issue, please try again later.")
-        else:
-            st.warning("Backend is still waking up. Please wait...")
-    else:
-        # Backend is already spun up, no need to show waiting message
-        pass
+    try:
+        response = requests.post(f"{API_URL}/start_registration", timeout=60)
+        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 = ""
+        st.session_state.summary = None
+        st.session_state.answer = ""
+        st.session_state.skip_address = False
+        st.session_state.skip_phone = False
+        st.session_state.prev_question = ""
+        st.session_state.question_number = 1
+    except requests.RequestException as e:
+        print(f"Error starting registration: {e}")
+        st.error(f"Error starting registration: {e}")
 
 def submit_response():
     if not st.session_state.session_id:
@@ -84,7 +64,7 @@ def submit_response():
         response.raise_for_status()
         data = response.json()
         print("API Response:", data)
-        if data.get("message") == "Registration complete!":
+        if data.get("message") == "Registration complete!"):
             st.session_state.summary = data["summary"]
             st.session_state.current_question = ""
             st.session_state.feedback = "Registration complete!"
@@ -99,13 +79,9 @@ def submit_response():
         st.session_state.skip_address = False
         st.session_state.skip_phone = False
         st.rerun()
-    except requests.exceptions.HTTPError as e:
-        if e.response.status_code == 403:
-            st.error("Backend unavailable, please try again later.")
-        else:
-            st.error(f"Unexpected error: {e}")
     except requests.RequestException as e:
-        st.error("Connection issue, please try again later.")
+        print(f"Error submitting response: {e}")
+        st.error(f"Error submitting response: {e}")
 
 def edit_field(field, value):
     if not st.session_state.session_id:
@@ -124,18 +100,14 @@ def edit_field(field, value):
         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":
+        if data.get("message") == "Needs clarification"):
             st.error(f"Clarification needed for {field}: {data['validation_feedback']}")
-        elif data.get("message") == "Field updated successfully!":
+        elif data.get("message") == "Field updated successfully!"):
             st.success("Database updated.")
             st.rerun()
-    except requests.exceptions.HTTPError as e:
-        if e.response.status_code == 403:
-            st.error("Backend unavailable, please try again later.")
-        else:
-            st.error(f"Unexpected error: {e}")
     except requests.RequestException as e:
-        st.error("Connection issue, please try again later.")
+        print(f"Error editing field: {e}")
+        st.error(f"Error editing field: {e}")
 
 def main():
     st.title("AI-Powered Registration System")
@@ -151,6 +123,7 @@ def main():
 
     # Tab 2: Registration Application
     with tab2:
+        st.subheader("Registration")
         registration_content = read_content_from_file("tab2.txt")
         st.markdown(registration_content)
         st.markdown("**Developed by [email protected]**")
@@ -165,13 +138,11 @@ def main():
                     edit_field(key, new_value)
             col1, col2 = st.columns(2)
             with col1:
-                if st.button("Next Registration", key="next_reg", on_click=lambda: start_registration()):
-                    st.session_state.session_id = None
-                    st.rerun()
+                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=lambda: start_registration()):
-                    st.session_state.session_id = None
-                    st.rerun()
+                if st.button("End Session", key="end_sess", on_click=start_registration):
+                    pass
         else:
             if st.session_state.get("current_question"):
                 if st.session_state.get("feedback"):
@@ -211,7 +182,7 @@ def main():
                 if st.button("Submit", key="submit_button"):
                     submit_response()
             else:
-                start_registration()  # Call start_registration if no current question
+                print("Waiting for session to initialize...")
 
 if __name__ == "__main__":
     # Initialize session state
@@ -226,6 +197,8 @@ if __name__ == "__main__":
         st.session_state.skip_phone = False
         st.session_state.prev_question = ""
         st.session_state.question_number = 1
-        st.session_state.backend_waiting = False
+
+    if st.session_state.session_id is None:
+        start_registration()
 
     main()