Update app.py

app.py

@@ -1,44 +1,42 @@
-import streamlit as st
 import openai
+import streamlit as st
+from tenacity import retry, stop_after_attempt, wait_fixed
 
-# Assuming your OpenAI API key is set correctly in your Hugging Face Secrets
+# Assuming your OpenAI API key is set in Hugging Face Spaces secrets
 openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-# Define the initial system message
-initial_messages = [
-    {
-        "role": "system",
-        "content": """You are an AI assistant that provides home staging recommendations for real estate agents. Your guidance helps highlight a property's best features by suggesting specific furniture arrangements, decor choices, and overall aesthetic themes to make homes more appealing to potential buyers. Consider the home's style, size, and unique features in your recommendations, aiming to create a welcoming and attractive environment that can lead to a successful sale."""
-    }
-]
+initial_messages_marketing = [{
+    "role": "system",
+    "content": """
+    Your detailed system message goes here...
+    """
+}]
 
-# Function to handle the chat interaction
-def custom_chat_gpt(home_details, location):
-    messages = initial_messages.copy()
-    user_input = f"I have a home for sale with these details: {home_details}. It's located in {location}. How should I stage it for potential buyers?"
+@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 CustomMarketingGPT(user_input, messages):
     messages.append({"role": "user", "content": user_input})
-    
-    try:
-        response = openai.ChatCompletion.create(
-            model="gpt-4",
-            messages=messages
-        )
-        chat_gpt_reply = response.choices[0].message["content"]
-        return chat_gpt_reply
-    except Exception as e:
-        st.error(f"An error occurred: {str(e)}")
-        return None
+    response = call_openai_api(messages)
+    marketing_reply = response.choices[0].message["content"]
+    messages.append({"role": "assistant", "content": marketing_reply})
+    return marketing_reply, messages
+
+def wrapped_marketing_gpt(area, home_details):
+    messages = initial_messages_marketing.copy()
+    user_input = f"I live in {area} and my home has these features: {home_details}. What are some creative marketing ideas to promote the sale of my home?"
+    reply, updated_messages = CustomMarketingGPT(user_input, messages)
+    return reply
 
-# Streamlit UI
-st.title("Home Staging Assistant")
-home_details = st.text_area("Home Details", placeholder="Describe the home's features, size, style, etc.")
-location = st.text_input("Location", placeholder="Enter the home's location or neighborhood.")
-generate_button = st.button('Get Staging Recommendations')
+# Streamlit Interface
+st.title("Home Marketing Assistant")
+area = st.text_input("Location")
+home_details = st.text_area("Home Details", placeholder="E.g., 4 bedrooms, renovated kitchen, hardwood floors...")
 
-if generate_button:
-    reply = custom_chat_gpt(home_details, location)
-    if reply:
-        st.markdown("### Staging Recommendations")
-        st.write(reply)
-    else:
-        st.write("An error occurred while generating recommendations. Please try again.")
+if st.button('Generate Marketing Ideas'):
+    reply = wrapped_marketing_gpt(area, home_details)
+    st.text_area("Marketing Ideas", reply, height=300)