Update app.py

app.py

@@ -1,65 +1,44 @@
 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"]
+openai.api_key = 'your_openai_api_key'
 
-initial_messages = [{
-    "role": "system",
-    "content": """
-        You are an assistant that helps people plan their first YouTube video. Your suggestions should be geared toward creating engaging content that matches the user's expertise, personality, and interests. Provide a structured plan including:
-        - Video Topic
-        - Suggested Title
-        - Video Outline (main points or segments)
-        - Hook/Opening Line
-        - Recommended Call to Action
-        """
-}]
-
-def call_openai_api(messages):
-    return openai.ChatCompletion.create(
+# 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",
-        messages=messages
+        prompt=prompt,
+        max_tokens=150
     )
-
-def CustomChatGPT(user_input, messages):
-    query = f"User's expertise and interests: {user_input}. Suggest a structured plan for their first YouTube video."
-    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
-
-# Set layout to wide
-st.set_page_config(layout="wide")
-
-# Centered title
-st.markdown("<h1 style='text-align: center; color: black;'>YouTube Video Planner</h1>", unsafe_allow_html=True)
-
-# Create columns for input and output
-col1, col2 = st.columns(2)
-
-with col1:
-    st.markdown("<h2 style='text-align: center; color: black;'>Your Expertise & Interests</h2>", unsafe_allow_html=True)
-    user_input = st.text_area("Describe your expertise and interests", placeholder="E.g., cooking, fitness, real estate, technology, travel, etc.")
-    generate_button = st.button('Generate Video Plan')
-
-if generate_button:
-    messages = initial_messages.copy()
-    reply, _ = CustomChatGPT(user_input, messages)
-
-    with col2:
-        st.markdown("<h2 style='text-align: center; color: black;'>Your YouTube Video Plan ⬇️</h2>", unsafe_allow_html=True)
-        st.write(reply)
-
-# Contact capture form
-st.markdown("<h2 style='text-align: center; color: black;'>Get in Touch for More Help ⬇️</h2>", unsafe_allow_html=True)
-with st.form(key='contact_form'):
-    name = st.text_input("Your Name", placeholder="Enter your name")
-    email = st.text_input("Your Email", placeholder="Enter your email address")
-    message = st.text_area("Your Message", placeholder="Let us know how we can assist you further")
-    submit_button = st.form_submit_button(label='Submit')
-
-    if submit_button:
-        st.success("Thank you for getting in touch! We'll get back to you shortly.")
+    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.")