Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import json
 
+# Function to validate zipcode
 def validate_zipcode(zipcode):
     try:
         if len(zipcode) == 5 and zipcode.isdigit():
@@ -10,24 +11,17 @@ def validate_zipcode(zipcode):
     except:
         return False
 
-st.title("Information Form")
-
-# Initialize session state
-if 'form_submitted' not in st.session_state:
-    st.session_state.form_submitted = False
-
-if not st.session_state.form_submitted:
-    with st.form(key='info_form'):
-        gender = st.radio("Gender", ("Male", "Female", "Other"))
-        city = st.radio("City", ("Oakland", "Berkeley", "SF"))
-        zipcode = st.text_input("Zipcode (Optional)", value="")
-        urgency = st.radio("Urgency", ("Immediate", "High", "Moderate", "General Inquiry"))
-        duration = st.radio("Duration", ("Long Term", "Transitional", "Temporary"))
-        needs = st.text_area("Needs (describe needed services + ideal qualities of shelter)")
-
-        submit_button = st.form_submit_button(label="Submit")
-
-    if submit_button:
+# Function to handle form submission
+@st.experimental_dialog("Fill out the form")
+def form_dialog():
+    gender = st.radio("Gender", ("Male", "Female", "Other"))
+    city = st.radio("City", ("Oakland", "Berkeley", "SF"))
+    zipcode = st.text_input("Zipcode (Optional)", value="")
+    urgency = st.radio("Urgency", ("Immediate", "High", "Moderate", "General Inquiry"))
+    duration = st.radio("Duration", ("Long Term", "Transitional", "Temporary"))
+    needs = st.text_area("Needs (describe needed services + ideal qualities of shelter)")
+
+    if st.button("Submit"):
         if zipcode and not validate_zipcode(zipcode):
             st.error("Please enter a valid 5-digit zipcode.")
         else:
@@ -39,15 +33,23 @@ if not st.session_state.form_submitted:
                 "Duration": duration,
                 "Needs": needs
             }
-
             with open('data.json', 'w') as f:
                 json.dump(data, f)
-
             st.session_state.form_submitted = True
-            st.success("Form submitted successfully!")
-            st.json(data)
+            st.session_state.data = data
+            st.rerun()
+
+# Initialize session state
+if 'form_submitted' not in st.session_state:
+    st.session_state.form_submitted = False
+
+if not st.session_state.form_submitted:
+    if st.button("Open Form"):
+        form_dialog()
 else:
-    st.success("Form already submitted.")
+    st.success("Form submitted successfully!")
+    st.json(st.session_state.data)
+
     with open('data.json', 'r') as f:
         data = json.load(f)
     st.json(data)