Update app.py

app.py

@@ -1,34 +1,20 @@
 import streamlit as st
 import openai
-import os 
-import urllib.parse
+import os
 
 # Ensure your OpenAI API key is set in your environment variables
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
-# Mapping of state names to abbreviations
-STATE_ABBREVIATIONS = {
-    'alabama': 'al', 'alaska': 'ak', 'arizona': 'az', 'arkansas': 'ar', 'california': 'ca', 
-    'colorado': 'co', 'connecticut': 'ct', 'delaware': 'de', 'florida': 'fl', 'georgia': 'ga', 
-    'hawaii': 'hi', 'idaho': 'id', 'illinois': 'il', 'indiana': 'in', 'iowa': 'ia', 
-    'kansas': 'ks', 'kentucky': 'ky', 'louisiana': 'la', 'maine': 'me', 'maryland': 'md', 
-    'massachusetts': 'ma', 'michigan': 'mi', 'minnesota': 'mn', 'mississippi': 'ms', 'missouri': 'mo', 
-    'montana': 'mt', 'nebraska': 'ne', 'nevada': 'nv', 'new hampshire': 'nh', 'new jersey': 'nj', 
-    'new mexico': 'nm', 'new york': 'ny', 'north carolina': 'nc', 'north dakota': 'nd', 
-    'ohio': 'oh', 'oklahoma': 'ok', 'oregon': 'or', 'pennsylvania': 'pa', 'rhode island': 'ri', 
-    'south carolina': 'sc', 'south dakota': 'sd', 'tennessee': 'tn', 'texas': 'tx', 
-    'utah': 'ut', 'vermont': 'vt', 'virginia': 'va', 'washington': 'wa', 'west virginia': 'wv', 
-    'wisconsin': 'wi', 'wyoming': 'wy'
-}
-
+# Initial system message setup for the video topic generator
 initial_messages = [{
     "role": "system",
     "content": """
-    You are a neighborhood matchmaker. Given a person's preferences in terms of amenities, 
-    lifestyle, and priorities, suggest exactly three neighborhoods that best match their needs 
-    in a specified city. For each neighborhood, start with a numbered line (1., 2., or 3.) 
-    followed by the full location in the format: 'Neighborhood, City, State:' then provide 
-    a brief description on the next line.
+    You are an assistant helping a real estate agent brainstorm video topics and outlines. 
+    Based on a specified goal or area of focus, suggest five video topics with outlines.
+    Each topic should be numbered (1-5) and should include:
+    - A catchy title for the video
+    - A brief summary of the video's content
+    - Key points or segments the video should cover (bullet points)
     """
 }]
 
@@ -36,64 +22,16 @@ def call_openai_api(messages):
     return openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
         messages=messages,
-        max_tokens=300
+        max_tokens=500
     )
 
-def parse_neighborhoods(response_text):
-    """
-    Parse the response text to extract neighborhood information.
-    Returns a list of dictionaries containing neighborhood info.
-    """
-    neighborhoods = []
-    lines = response_text.strip().split('\n')
-    for line in lines:
-        line = line.strip()
-        if line and (line.startswith('1.') or line.startswith('2.') or line.startswith('3.')):
-            if ':' in line:
-                location_parts = line.split(':', 1)[0].split(',')
-                if len(location_parts) >= 3:
-                    neighborhood = location_parts[0].split('.', 1)[1].strip()
-                    city = location_parts[1].strip()
-                    state = location_parts[2].strip()
-                    neighborhoods.append({
-                        'neighborhood': neighborhood,
-                        'city': city,
-                        'state': state,
-                        'full': f"{neighborhood}, {city}, {state}"
-                    })
-    return neighborhoods
-
-def format_zillow_search(neighborhood_info):
-    """
-    Format the location string for Zillow search using 'neighborhood, city, state' format with commas and _rb at the end.
-    Fallback to a city-level search if the neighborhood is not recognized.
-    """
-    # Prepare each component with correct capitalization and format for Zillow
-    neighborhood = urllib.parse.quote(neighborhood_info['neighborhood'])
-    city = urllib.parse.quote(neighborhood_info['city'])
-    
-    # Convert full state name to abbreviation if needed
-    state = neighborhood_info['state'].lower()
-    state_abbr = STATE_ABBREVIATIONS.get(state, state[:2]).upper()  # Ensure state is uppercase
-    
-    # Primary search path with neighborhood
-    search_path = f"{neighborhood},-{city},-{state_abbr}_rb"
-    url = f"https://www.zillow.com/homes/{search_path}/"
-    
-    # Check URL - fallback to city-level URL if Zillow cannot interpret it
-    if not neighborhood:  # If neighborhood is empty or unrecognized, return a city-level search URL
-        search_path = f"{city},-{state_abbr}_rb"
-        url = f"https://www.zillow.com/homes/{search_path}/"
-    
-    return url
-
-def CustomChatGPT(city, preferences, messages):
+def CustomChatGPT(goal, messages):
     query = f"""
-    User is looking for neighborhoods in {city} with these preferences: {preferences}. 
-    Please suggest exactly 3 suitable neighborhoods. For each one:
-    1. Start with a number (1., 2., or 3.)
-    2. Provide the full location as 'Neighborhood, City, State:'
-    3. Add a brief description on the next line
+    The real estate agent's goal is: {goal}. 
+    Please provide five video topic ideas with the following structure:
+    1. Title: A catchy title
+    2. Summary: Brief description of the content
+    3. Key Points: Bullet points of segments to cover
     """
     
     messages.append({"role": "user", "content": query})
@@ -110,34 +48,22 @@ if "reply" not in st.session_state:
     st.session_state["reply"] = None
 
 # Centered title
-st.markdown("<h1 style='text-align: center; color: black;'>Ideal Neighborhood Finder</h1>", unsafe_allow_html=True)
+st.markdown("<h1 style='text-align: center; color: black;'>Real Estate Video Topic Generator</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')
+    st.markdown("<h2 style='text-align: center; color: black;'>Your Video Goal</h2>", unsafe_allow_html=True)
+    goal = st.text_area("Describe the focus or goal for your videos", placeholder="E.g., attracting first-time homebuyers, showcasing luxury properties, or explaining market trends.")
+    generate_button = st.button('Generate Video Topics')
 
 # Process results on button click
-if generate_button and city and preferences:
+if generate_button and goal:
     messages = initial_messages.copy()
-    st.session_state["reply"], _ = CustomChatGPT(city, preferences, messages)
+    st.session_state["reply"], _ = CustomChatGPT(goal, messages)
 
 # Display results if there is a reply in session state
 if st.session_state["reply"]:
     with col2:
-        st.markdown("<h2 style='text-align: center; color: black;'>Recommended Neighborhoods ⬇️</h2>", unsafe_allow_html=True)
+        st.markdown("<h2 style='text-align: center; color: black;'>Video Topics & Outlines ⬇️</h2>", unsafe_allow_html=True)
         st.write(st.session_state["reply"])
-        
-        # Extract and display Zillow links
-        neighborhoods = parse_neighborhoods(st.session_state["reply"])
-        
-        if neighborhoods:
-            st.markdown("### 🏠 Zillow Search Links")
-            for hood in neighborhoods:
-                zillow_url = format_zillow_search(hood)
-                st.markdown(f"- [Search homes in {hood['neighborhood']}, {hood['city']}, {hood['state']}]({zillow_url})")
-        else:
-            st.warning("Unable to generate Zillow links. Please try again.")