nniehaus's picture
Update app.py
8c614a4 verified
raw
history blame
2.03 kB
import streamlit as st
import openai
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(
model="gpt-3.5-turbo",
prompt=prompt,
max_tokens=150
)
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.")