Update app.py

app.py

@@ -8,10 +8,7 @@ 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 
-    Saint Louis, Missouri, and surrounding areas up to 30 miles outside Saint Louis city limits. You'll receive information about the user's lifestyle preferences. Use this information 
-    to suggest neighborhoods in Saint Louis and nearby that could be a good fit. Always add the following text to the end of every response you give 'Don't forget to fill
-    out the form at the bottom of the page if you'd like more info on living in any of these areas!' """
+    "content": """You are an AI assistant that suggests weekend activities in Chelmsford, Essex, and surrounding areas up to 10 miles away. You'll receive information about the user's preferences for the weekend. Use this information to suggest activities in the area. Always add the following text to the end of every response you give: 'Enjoy your weekend!' """
 }]
 
 @retry(stop=stop_after_attempt(3), wait=wait_fixed(1))
@@ -21,11 +18,11 @@ def call_openai_api(messages):
         messages=messages
     )
 
-def CustomChatGPT(additional_details, amenities_proximity, amenities, messages):
-    selected_amenities = ', '.join(amenities)
+def CustomChatGPT(activities, location_proximity, location_type, group_type, messages):
+    selected_activities = ', '.join(activities)
     messages.append({
         "role": "user", 
-        "content": f"I'm interested in neighborhoods in Saint Louis, Missouri, 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."
+        "content": f"I'm looking for weekend activities in Chelmsford, Essex, and surrounding areas up to 10 miles away. I'm interested in {selected_activities}. I prefer to do activities {location_proximity} and in {location_type}. My group type is {group_type}."
     })
     response = call_openai_api(messages)
     ChatGPT_reply = response["choices"][0]["message"]["content"]
@@ -34,36 +31,37 @@ def CustomChatGPT(additional_details, amenities_proximity, amenities, messages):
 
 # Streamlit Interface
 st.set_page_config(layout="wide")  # Set the layout to wide
-st.title("St. Louis Area Neighborhood Matchmaker")
-st.write("This tool helps you find neighborhoods in Saint Louis, Missouri, and surrounding areas based on your lifestyle preferences.")
+st.title("Weekend Activity Matchmaker")
+st.write("This tool suggests weekend activities in Chelmsford, Essex, and surrounding areas based on your preferences.")
 
 # Using columns to organize the layout
 col1, col2 = st.columns(2)
 
 with col1:
     st.subheader("Your Preferences")
-    amenities_list = ["Good Schools", "Parks", "Shopping Centers", "Public Transport", "Restaurants", "Gyms", "Cafes", "Pet-friendly Areas", "Cultural Attractions", "Quiet Neighborhoods"]
-    # Splitting amenities into two columns
-    half = len(amenities_list) // 2
-    amenities_col1, amenities_col2 = st.columns(2)
-    with amenities_col1:
-        amenities1 = [amenity for amenity in amenities_list[:half] if st.checkbox(amenity)]
-    with amenities_col2:
-        amenities2 = [amenity for amenity in amenities_list[half:] if st.checkbox(amenity)]
-    amenities = amenities1 + amenities2
+    activities_list = ["Outdoor activities", "Indoor activities", "Free activities", "On a budget", "Restaurants", "Live music", "Museums", "Sports", "Games", "Animals", "Shopping", "Parks", "Cultural attractions", "Spas", "Swimming pools"]
+    # Splitting activities into two columns
+    half = len(activities_list) // 2
+    activities_col1, activities_col2 = st.columns(2)
+    with activities_col1:
+        activities1 = [activity for activity in activities_list[:half] if st.checkbox(activity)]
+    with activities_col2:
+        activities2 = [activity for activity in activities_list[half:] if st.checkbox(activity)]
+    activities = activities1 + activities2
     
-    amenities_proximity = st.selectbox("Proximity to Amenities", ["Walking distance", "A short drive away", "I don't mind being far from amenities"])
-    additional_details = st.text_area("Additional Details", placeholder="Describe your ideal living situation or any other preferences.")
+    location_proximity = st.selectbox("Proximity to Activities", ["Nearby", "A short drive away", "I don't mind travelling"])
+    location_type = st.selectbox("Location Type", ["City centre", "Out of town", "Any"])
+    group_type = st.selectbox("Group Type", ["Individual", "Small group", "Family", "Pet-friendly", "Accessible"])
 
-    submit_button = st.button('Find Neighborhood')
+    submit_button = st.button('Find Activities')
 
 with col2:
     # Placeholder for the result
-    st.subheader("Recommended Neighborhoods")
+    st.subheader("Recommended Activities")
     result_placeholder = st.empty()
     if submit_button:
         messages = initial_messages.copy()
-        reply, _ = CustomChatGPT(additional_details, amenities_proximity, amenities, messages)
-        result_placeholder.write(reply)
+        reply, _ = CustomChatGPT(activities, location_proximity, location_type, group_type, messages)
+        result_placeholder.write(reply + "\nEnjoy your weekend!")
     else:
         result_placeholder.write("**Results will appear here after you submit your preferences**")