Upload index.py

index.py

@@ -28,30 +28,41 @@ def start_registration():
     print("Starting registration...")
     if "backend_waiting" not in st.session_state:
         st.session_state.backend_waiting = False
-    if not st.session_state.backend_waiting:
-        st.session_state.backend_waiting = True
-        st.warning("Waiting up to 60secs 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
-        except requests.RequestException as e:
-            print(f"Error starting registration: {e}")
-            st.error(f"Error starting registration: {e}")
-            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:
-        st.warning("Backend is still waking up. Please wait...")
+        # Backend is already spun up, no need to show waiting message
+        pass
 
 def submit_response():
     if not st.session_state.session_id:
@@ -88,9 +99,13 @@ 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:
-        print(f"Error submitting response: {e}")
-        st.error(f"Error submitting response: {e}")
+        st.error("Connection issue, please try again later.")
 
 def edit_field(field, value):
     if not st.session_state.session_id:
@@ -111,12 +126,16 @@ def edit_field(field, value):
         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!":
+        elif data.get("message") == "Field updated successfully!"):
             st.success("Database updated.")
-            st.run()
+            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:
-        print(f"Error editing field: {e}")
-        st.error(f"Error editing field: {e}")
+        st.error("Connection issue, please try again later.")
 
 def main():
     st.title("AI-Powered Registration System")
@@ -192,7 +211,7 @@ def main():
                 if st.button("Submit", key="submit_button"):
                     submit_response()
             else:
-                print("Waiting for session to initialize...")
+                start_registration()  # Call start_registration if no current question
 
 if __name__ == "__main__":
     # Initialize session state
@@ -209,7 +228,4 @@ if __name__ == "__main__":
         st.session_state.question_number = 1
         st.session_state.backend_waiting = False
 
-    if st.session_state.session_id is None:
-        start_registration()
-
     main()