Update app.py

app.py

@@ -1,25 +1,19 @@
 import streamlit as st
 import openai
 import os
-import folium
-from streamlit_folium import st_folium
-import urllib.parse
-import subprocess
 
 # Ensure your OpenAI API key is set in your environment variables
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
-# Ensure folium is installed
-try:
-    import folium
-except ModuleNotFoundError:
-    subprocess.check_call(["python", "-m", "pip", "install", "folium"])
-    import folium
-
 initial_messages = [{
     "role": "system",
     "content": """
-        You are a neighborhood matchmaker. Given a person's preferences in terms of amenities, lifestyle, and priorities, suggest three neighborhoods that best match their needs. Provide a short description for each neighborhood, including key features such as nearby amenities, atmosphere, and suitability for the user's preferences.
+        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
         """
 }]
 
@@ -29,72 +23,38 @@ def call_openai_api(messages):
         messages=messages
     )
 
-def CustomChatGPT(preferences, messages):
-    query = f"User preferences: {preferences}. Suggest suitable neighborhoods."
+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']
+    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")
 
-# Initialize session state for storing the results
-if "reply" not in st.session_state:
-    st.session_state.reply = ""
-
 # Centered title
-st.markdown("<h1 style='text-align: center; color: black;'>Neighborhood Matchmaker</h1>", unsafe_allow_html=True)
+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 Preferences</h2>", unsafe_allow_html=True)
-    preferences = st.text_area("Describe your ideal neighborhood", placeholder="E.g., family-friendly, good schools, near parks, vibrant nightlife, public transportation, etc.")
-    generate_button = st.button('Find Neighborhoods')
+    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(preferences, messages)
-    st.session_state.reply = reply  # Store the reply in session state
+    reply, _ = CustomChatGPT(user_input, messages)
 
-if st.session_state.reply:
     with col2:
-        st.markdown("<h2 style='text-align: center; color: black;'>Recommended Neighborhoods ⬇️</h2>", unsafe_allow_html=True)
-        st.write(st.session_state.reply)
-
-        # Add map integration
-        st.markdown("<h2 style='text-align: center; color: black;'>Map of Suggested Neighborhoods ⬇️</h2>", unsafe_allow_html=True)
-        map_center = [37.7749, -122.4194]  # Default to San Francisco coordinates
-        m = folium.Map(location=map_center, zoom_start=12)
-
-        # Placeholder coordinates for neighborhoods (these should be updated with actual data)
-        neighborhoods = [
-            {"name": "Neighborhood 1", "coordinates": [37.7749, -122.4194]},
-            {"name": "Neighborhood 2", "coordinates": [37.7849, -122.4094]},
-            {"name": "Neighborhood 3", "coordinates": [37.7649, -122.4294]}
-        ]
-
-        for neighborhood in neighborhoods:
-            folium.Marker(
-                location=neighborhood["coordinates"],
-                popup=neighborhood["name"],
-                icon=folium.Icon(color='blue')
-            ).add_to(m)
-
-        st_folium(m, width=700, height=500)
-
-        # Add Zillow search links for neighborhoods
-        st.markdown("<h2 style='text-align: center; color: black;'>Search for Homes in Suggested Neighborhoods ⬇️</h2>", unsafe_allow_html=True)
-        for neighborhood in neighborhoods:
-            query = urllib.parse.quote(neighborhood["name"])
-            zillow_url = f"https://www.zillow.com/homes/{query}_rb/"
-            st.markdown(f"[Search for homes in {neighborhood['name']} on Zillow]({zillow_url})")
+        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 Details ⬇️</h2>", unsafe_allow_html=True)
+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")