app updated

app.py

@@ -40,6 +40,12 @@ def form_dialog():
             st.session_state.data = data
             st.rerun()
 
+# Function to reset session state
+def reset_session_state():
+    st.session_state.form_submitted = False
+    st.session_state.shelter_index = 0
+    st.experimental_rerun()
+
 # Initialize session state
 if 'form_submitted' not in st.session_state:
     st.session_state.form_submitted = False
@@ -110,38 +116,10 @@ else:
     if len(shelters) == 0:
         st.warning("No shelters found matching your criteria.")
     else:
-        # Debug print to see the structure of the first shelter
-        # st.write("Shelter structure:", shelters[0])
-
         # Create two columns
         col1, col2 = st.columns([1, 2])
 
         with col1:
-            # Custom HTML for buttons
-            # st.markdown("""
-            #     <style>
-            #     .button-container {
-            #         display: flex;
-            #         flex-direction: column;
-            #         align-items: center;
-            #         gap: 20px;
-            #         width: 100%;
-            #     }
-            #     .stButton button {
-            #         background-color: #4CAF50;
-            #         color: white;
-            #         padding: 10px 20px;
-            #         border: none;
-            #         cursor: pointer;
-            #         font-size: 16px;
-            #     }
-            #     </style>
-            #     <div class="button-container">
-            #         <button onclick="document.querySelector('button[aria-label=Previous]').click()">Previous</button>
-            #         <button onclick="document.querySelector('button[aria-label=Next]').click()">Next</button>
-            #     </div>
-            #     """, unsafe_allow_html=True)
-
             if st.button("Previous", key="previous"):
                 if st.session_state.shelter_index > 0:
                     st.session_state.shelter_index -= 1
@@ -152,11 +130,12 @@ else:
                     st.session_state.shelter_index += 1
                     st.experimental_rerun()
 
+            if st.button("Resubmit", key="resubmit"):
+                reset_session_state()
+
         with col2:
             # Display the current shelter information
             shelter = shelters[st.session_state.shelter_index]
-            some_random_name = f'### {shelter["description"]}'
-            st.markdown(some_random_name)
             st.write(shelter["description"])
             st.header(shelter["header"])
             st.image(shelter["image_url"])