Update app.py

app.py

@@ -1,48 +1,50 @@
-import streamlit as st 
+import streamlit as st
 import openai
 import os
-
-# Ensure your OpenAI API key is set in your environment variables
-openai.api_key = os.environ["YOUR_OPENAI_API_KEY"]
-
-initial_messages = [{"role": "system", "content": """You are an AI assistant that helps real estate professionals create a personalized video marketing roadmap. Based on the user's business profile, video experience, resources, and goals, generate a detailed plan covering the four key strategies: One-to-One Video Message Habit, Scaling with Video, Video Marketing Funnel, and Leveraging Social Media."""}]
-
+from tenacity import retry, wait_fixed, stop_after_attempt
+
+# Set the OpenAI API key
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+# Define the initial system message
+initial_messages = [
+    {
+        "role": "system",
+        "content": """You are an AI assistant that provides highly specific investment property recommendations in Breckenridge and 
+        Blue River, CO. Consider the user's maximum price, selected amenities, and area preferences to recommend specific neighborhoods 
+        or intersections that would be a good fit for investment. Importantly, only recommend areas where short-term rentals are allowed 
+        under the new law."""
+    }
+]
+
+# Add retry logic to the API call
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(1))
 def call_openai_api(messages):
-    return openai.ChatCompletion.create(
+    response = openai.ChatCompletion.create(
         model="gpt-4",
         messages=messages
     )
+    return response
 
-def CustomChatGPT(user_input, messages):
+# Function to handle the chat interaction
+def InvestmentPropertyFinder(max_price, amenities, bedrooms, bathrooms, square_footage, messages):
+    user_input = f"I'm considering buying an investment property in Breckenridge and Blue River, CO. My maximum price is {max_price}. I'm looking for these amenities: {', '.join(amenities)}. Bedrooms: {bedrooms}, Bathrooms: {bathrooms}, Square Footage: {square_footage if square_footage else 'No preference'}."
     messages.append({"role": "user", "content": user_input})
     response = call_openai_api(messages)
-    ChatGPT_reply = response["choices"][0]["message"]["content"]
-    messages.append({"role": "assistant", "content": ChatGPT_reply})
-    return ChatGPT_reply, messages
+    finder_reply = response.choices[0].message.content
+    messages.append({"role": "assistant", "content": finder_reply})
+    return finder_reply, messages
 
 # Streamlit UI
-st.title("Small Business Video Roadmap Tool for Realtors")
-st.write("This tool generates a personalized video marketing roadmap for real estate professionals. Fill in the details about your business, video experience, resources, and goals.")
-
-# Collecting user inputs
-business_type = st.text_input("Type of Real Estate Business", placeholder="e.g., Residential, Commercial, Rental")
-target_audience = st.text_input("Target Audience Demographics", placeholder="e.g., First-time Homebuyers, Luxury Clients")
-current_marketing = st.text_input("Current Marketing Strategies", placeholder="e.g., Social Media, Email Campaigns")
-
-video_experience = st.selectbox("Video Production Skills Level", ["Beginner", "Intermediate", "Advanced"])
-video_marketing_experience = st.selectbox("Experience with Video Marketing", ["None", "Some", "Extensive"])
-
-budget = st.text_input("Budget for Video Production", placeholder="Enter your budget")
-time_availability = st.text_input("Time Availability for Video Creation", placeholder="e.g., 5 hours a week")
-equipment = st.text_input("Available Equipment for Video Production", placeholder="e.g., Smartphone, DSLR Camera")
-
-goals = st.text_area("Video Marketing Goals", placeholder="Enter your specific objectives for using video in your business")
-desired_timeline = st.text_input("Desired Timeline for Achieving Goals", placeholder="e.g., 6 months")
+st.title("Investment Property Finder in Breckenridge and Blue River, CO")
 
-submit_button = st.button('Generate Video Roadmap')
+max_price = st.text_input("Maximum Price", "Enter your maximum price for the property.")
+amenities = st.multiselect("Amenities", ["Close to Skiing", "Hot Tub", "Bus Route", "Parking"], "Select Amenities")
+bedrooms = st.selectbox("Bedrooms", ["1", "2", "3", "4", "5+"])
+bathrooms = st.selectbox("Bathrooms", ["1", "2", "3", "4", "5+"])
+square_footage = st.selectbox("Square Footage", ["<1000 sqft", "1000-2000 sqft", "2000-3000 sqft", "3000-4000 sqft", "4000+ sqft"])
 
-if submit_button:
+if st.button('Find Properties'):
     messages = initial_messages.copy()
-    user_input = f"My real estate business focuses on {business_type} targeting {target_audience}. I currently use {current_marketing} for marketing. My video production skills are {video_experience}, and my experience with video marketing is {video_marketing_experience}. My budget for video production is {budget}, and I can allocate {time_availability} for video creation using {equipment}. My goals are {goals}, and I want to achieve these in {desired_timeline}."
-    reply, _ = CustomChatGPT(user_input, messages)
+    reply, updated_messages = InvestmentPropertyFinder(max_price, amenities, bedrooms, bathrooms, square_footage, messages)
     st.write(reply)