Update app.py

app.py

@@ -1,44 +1,60 @@
 import streamlit as st
 import openai
+import os
 import urllib.parse
 
 # Ensure your OpenAI API key is set in your environment variables
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
-# Function to get neighborhood suggestions from OpenAI
-def suggest_neighborhoods(user_preferences, city=""):
-    prompt = (f"Based on the following preferences, suggest neighborhoods in {city}: {user_preferences}. "
-              "For each neighborhood, provide a brief description including its main attractions and why it would suit these preferences.")
-    response = openai.Completion.create(
+initial_messages = [{
+    "role": "system",
+    "content": """
+        You are a neighborhood matchmaker. Given a person's preferences in terms of amenities, lifestyle, and priorities, suggest three neighborhoods that best match their needs in a specified city. Provide a short description of each neighborhood, highlighting its unique qualities.
+    """
+}]
+
+def call_openai_api(messages):
+    return openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
-        prompt=prompt,
-        max_tokens=150
+        messages=messages
     )
-    return response.choices[0].text.strip()
-
-# Function to generate Zillow search link for a neighborhood
-def generate_zillow_link(neighborhood):
-    neighborhood_query = urllib.parse.quote(neighborhood)
-    return f"https://www.zillow.com/homes/{neighborhood_query}_rb/"
-
-# Streamlit Interface
-st.title("Ideal Neighborhood Finder")
-st.write("Describe what you're looking for in an ideal neighborhood, and we'll suggest neighborhoods with Zillow links for homes in those areas.")
-
-# User Inputs
-city = st.text_input("City", placeholder="Enter the city you want to search in")
-user_preferences = st.text_area("Describe Your Ideal Neighborhood", placeholder="E.g., close to schools, parks, public transit, vibrant nightlife, etc.")
-
-if st.button("Find Neighborhoods"):
-    if city and user_preferences:
-        # Get neighborhood suggestions from OpenAI
-        neighborhoods = suggest_neighborhoods(user_preferences, city)
-        
-        # Display results
-        st.subheader("Neighborhood Suggestions")
-        for neighborhood in neighborhoods.splitlines():
-            if neighborhood:  # Filter out empty lines
-                zillow_link = generate_zillow_link(neighborhood)
-                st.markdown(f"- **{neighborhood}**: [View homes on Zillow]({zillow_link})")
-    else:
-        st.error("Please provide both a city and a description of your ideal neighborhood.")
+
+def CustomChatGPT(city, preferences, messages):
+    query = f"User is looking for neighborhoods in {city} with these preferences: {preferences}. Suggest suitable neighborhoods and describe each briefly."
+    messages.append({"role": "user", "content": query})
+    response = call_openai_api(messages)
+    ChatGPT_reply = response["choices"][0]["message"]["content"]
+    messages.append({"role": "assistant", "content": ChatGPT_reply})
+    return ChatGPT_reply, messages
+
+# Streamlit setup
+st.set_page_config(layout="wide")
+
+# Centered title
+st.markdown("<h1 style='text-align: center; color: black;'>Ideal Neighborhood Finder</h1>", unsafe_allow_html=True)
+
+# User inputs
+col1, col2 = st.columns(2)
+
+with col1:
+    st.markdown("<h2 style='text-align: center; color: black;'>Your Preferences</h2>", unsafe_allow_html=True)
+    city = st.text_input("City", placeholder="Enter the city you want to search in")
+    preferences = st.text_area("Describe your ideal neighborhood", placeholder="E.g., family-friendly, near parks, vibrant nightlife")
+    generate_button = st.button('Find Neighborhoods')
+
+# Process results on button click
+if generate_button and city and preferences:
+    messages = initial_messages.copy()
+    reply, _ = CustomChatGPT(city, preferences, messages)
+
+    # Display the results
+    with col2:
+        st.markdown("<h2 style='text-align: center; color: black;'>Recommended Neighborhoods ⬇️</h2>", unsafe_allow_html=True)
+        st.write(reply)
+
+        # Display Zillow links for each neighborhood
+        neighborhoods = reply.splitlines()
+        for neighborhood in neighborhoods:
+            neighborhood_query = urllib.parse.quote(neighborhood)
+            zillow_url = f"https://www.zillow.com/homes/{neighborhood_query}_rb/"
+            st.markdown(f"- [Search for homes in {neighborhood} on Zillow]({zillow_url})")