Update app.py

app.py

@@ -10,48 +10,44 @@ def validate_zipcode(zipcode):
     except:
         return False
 
-def main():
-    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:
-            if zipcode and not validate_zipcode(zipcode):
-                st.error("Please enter a valid 5-digit zipcode.")
-            else:
-                data = {
-                    "Gender": gender,
-                    "City": city,
-                    "Zipcode": zipcode if zipcode else "N/A",
-                    "Urgency": urgency,
-                    "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)
-    else:
-        st.success("Form already submitted.")
-        with open('data.json', 'r') as f:
-            data = json.load(f)
-        st.json(data)
-
-if __name__ == '__main__':
-    main()
+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:
+        if zipcode and not validate_zipcode(zipcode):
+            st.error("Please enter a valid 5-digit zipcode.")
+        else:
+            data = {
+                "Gender": gender,
+                "City": city,
+                "Zipcode": zipcode if zipcode else "N/A",
+                "Urgency": urgency,
+                "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)
+else:
+    st.success("Form already submitted.")
+    with open('data.json', 'r') as f:
+        data = json.load(f)
+    st.json(data)