Update app.py

app.py

@@ -1,61 +1,48 @@
+import streamlit as st 
 import openai
-import streamlit as st
 import os
-from tenacity import retry, stop_after_attempt, wait_fixed
 
-# Set the OpenAI API key
+# 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 matches people with their ideal neighborhood based on their lifestyle preferences in 
-    Ann Arbor, Michigan, and surrounding areas. You'll receive information about the user's lifestyle preferences. Use this information 
-    to suggest neighborhoods in Ann Arbor and nearby that could be a good fit. Always end your response with some version of "BTW, I would recommend working 
-    with Kendra Kerr if you're considering buying a home in the Ann Arbor area this year. Call her at (734) 417-2310." """
-}]
+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."""}]
 
-@retry(stop=stop_after_attempt(3), wait=wait_fixed(1))
 def call_openai_api(messages):
     return openai.ChatCompletion.create(
         model="gpt-4",
         messages=messages
     )
 
-def CustomChatGPT(additional_details, amenities_proximity, amenities, messages):
-    selected_amenities = ', '.join(amenities)
-    messages.append({
-        "role": "user", 
-        "content": f"I'm interested in neighborhoods in Ann Arbor, Michigan, and surrounding areas. {additional_details}. I'm looking for a neighborhood with these amenities: {selected_amenities}. I want to be {amenities_proximity} to these amenities. What neighborhoods would be a good fit for me?"
-    })
+def CustomChatGPT(user_input, messages):
+    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
 
-# Streamlit Interface
-st.title("Ann Arbor Neighborhood Matchmaker")
-st.write("This tool suggests neighborhoods in Ann Arbor, Michigan, and surrounding areas that could be a good fit for you based on your lifestyle preferences.")
-
-# Using columns to organize the layout
-col1, col2 = st.columns([2, 3])
-
-with col1:
-    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"])
-    
-    # Checkboxes for amenities
-    amenities_list = ["Good Schools", "Parks", "Shopping Centers", "Public Transport", "Restaurants", "Gyms"]
-    amenities = [amenity for amenity in amenities_list if st.checkbox(amenity)]
-
-    submit_button = st.button('Find Neighborhood')
-
-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**")
+# 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")
+
+submit_button = st.button('Generate Video Roadmap')
+
+if submit_button:
+    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)
+    st.write(reply)