Update app/building_info_form.py

app/building_info_form.py

@@ -130,7 +130,10 @@ class BuildingInfoForm:
                 help="Enter the total floor area of the building in square meters (optional if width and depth provided)"
             )
             
-            st.markdown("OR")  # Added OR between area and width/depth
+            # Center the OR using columns
+            col1, col2, col3 = st.columns([2, 1, 2])
+            with col2:
+                st.markdown("OR")
             
             col1, col2 = st.columns(2)
             with col1:
@@ -150,7 +153,7 @@ class BuildingInfoForm:
                     help="Enter the building's depth in meters (optional if area provided)"
                 )
             
-            st.subheader("Building Orientation")  # Changed to subheader
+            st.subheader("Building Orientation")
             session_state["building_info"]["orientation"] = st.selectbox(
                 "Building Orientation",
                 ["NORTH", "NORTHEAST", "EAST", "SOUTHEAST", "SOUTH", "SOUTHWEST", "WEST", "NORTHWEST"],
@@ -158,7 +161,7 @@ class BuildingInfoForm:
                 help="Select the direction of the building's main facade"
             )
             
-            st.subheader("Operating Hours")  # Changed to subheader
+            st.subheader("Operating Hours")
             session_state["building_info"]["operating_hours"] = st.text_input(
                 "Operating Hours",
                 value=session_state["building_info"]["operating_hours"],
@@ -168,23 +171,18 @@ class BuildingInfoForm:
             submitted = st.form_submit_button("Save Building Information")
             
             if submitted:
-                # Validate before saving
                 valid, errors = self.validate_building_info(session_state["building_info"])
                 if not valid:
                     for error in errors:
                         st.error(error)
                 else:
-                    # Logic for handling dimensions
                     if session_state["building_info"]["width"] > 0 and session_state["building_info"]["depth"] > 0:
                         calculated_area = session_state["building_info"]["width"] * session_state["building_info"]["depth"]
-                        # If no area provided or area is 0, use calculated area
                         if session_state["building_info"]["floor_area"] == 0:
                             session_state["building_info"]["floor_area"] = calculated_area
-                        # If area is provided, keep it
                     
                     total_volume = session_state["building_info"]["floor_area"] * session_state["building_info"]["building_height"]
                     
-                    # Store results in session state to persist across rerenders
                     session_state["save_results"] = {
                         "success": "Building information saved successfully!",
                         "area": f"Total Floor Area: {session_state['building_info']['floor_area']:.1f} m²",
@@ -192,20 +190,19 @@ class BuildingInfoForm:
                     }
                     session_state["data_saved"] = True
         
-        # Display saved results if they exist
+        # Display results if they exist
         if "save_results" in session_state and session_state["data_saved"]:
             st.success(session_state["save_results"]["success"])
             st.info(session_state["save_results"]["area"])
             st.info(session_state["save_results"]["volume"])
         
-        # Separate proceed button
+        # Proceed button with immediate navigation
         if session_state["data_saved"]:
             if st.button("Proceed to Climate Data"):
                 session_state["page"] = "Climate Data"
-                # Clear saved state to reset form
+                session_state["data_saved"] = False
                 if "save_results" in session_state:
                     del session_state["save_results"]
-                session_state["data_saved"] = False
     
     @staticmethod
     def validate_building_info(building_info: Dict[str, Any]) -> Tuple[bool, List[str]]: