Update app.py

app.py

@@ -3,13 +3,6 @@ import streamlit as st
 import os
 from tenacity import retry, stop_after_attempt, wait_fixed
 
-# Set the OpenAI API key
-openai.api_key = os.environ["YOUR_OPENAI_API_KEY"]
-import openai
-import streamlit as st
-import os
-from tenacity import retry, stop_after_attempt, wait_fixed
-
 # Set the OpenAI API key
 openai.api_key = os.environ["YOUR_OPENAI_API_KEY"]
 
@@ -44,40 +37,26 @@ st.set_page_config(layout="wide")  # Set the layout to wide
 st.title("St. Louis Area Neighborhood Matchmaker")
 st.write("This tool helps you find neighborhoods in Saint Louis, Missouri, and surrounding areas based on your lifestyle preferences.")
 
-# Checkboxes for amenities
-amenities_list = ["Good Schools", "Parks", "Shopping Centers", "Public Transport", "Restaurants", "Gyms", "Cafes", "Pet-friendly Areas", "Cultural Attractions", "Quiet Neighborhoods"]
-amenities = []
+# Using columns to organize the layout
+col1, col2 = st.columns([1, 2])
 
-# Organize checkboxes into columns
-col1, col2, col3, col4, col5 = st.columns(5)
 with col1:
-    for amenity in amenities_list[:2]:
-        if st.checkbox(amenity, key=amenity):
-            amenities.append(amenity)
-with col2:
-    for amenity in amenities_list[2:4]:
-        if st.checkbox(amenity, key=amenity):
-            amenities.append(amenity)
-with col3:
-    for amenity in amenities_list[4:6]:
-        if st.checkbox(amenity, key=amenity):
-            amenities.append(amenity)
-with col4:
-    for amenity in amenities_list[6:8]:
-        if st.checkbox(amenity, key=amenity):
-            amenities.append(amenity)
-with col5:
-    for amenity in amenities_list[8:]:
-        if st.checkbox(amenity, key=amenity):
-            amenities.append(amenity)
+    # Checkboxes for amenities
+    amenities_list = ["Good Schools", "Parks", "Shopping Centers", "Public Transport", "Restaurants", "Gyms", "Cafes", "Pet-friendly Areas", "Cultural Attractions", "Quiet Neighborhoods"]
+    amenities = [amenity for amenity in amenities_list if st.checkbox(amenity)]
 
-amenities_proximity = st.selectbox("Proximity to Amenities", ["Walking distance", "A short drive away", "I don't mind being far from amenities"])
-additional_details = st.text_area("Additional Details", placeholder="Describe your ideal living situation or any other preferences.")
+    amenities_proximity = st.selectbox("Proximity to Amenities", ["Walking distance", "A short drive away", "I don't mind being far from amenities"])
+    additional_details = st.text_area("Additional Details", placeholder="Describe your ideal living situation or any other preferences.")
+    
+    submit_button = st.button('Find Neighborhood')
 
-submit_button = st.button('Find Neighborhood')
-
-if submit_button:
-    messages = initial_messages.copy()
-    reply, _ = CustomChatGPT(additional_details, amenities_proximity, amenities, messages)
-    st.markdown("**Recommended Neighborhoods:**")
-    st.write(reply)
+with col2:
+    # Placeholder for the result
+    result_placeholder = st.empty()
+    if submit_button:
+        messages = initial_messages.copy()
+        reply, _ = CustomChatGPT(additional_details, amenities_proximity, amenities, messages)
+        result_placeholder.markdown("**Recommended Neighborhoods:**")
+        result_placeholder.write(reply)
+    else:
+        result_placeholder.write("**Results will appear here**")