Update app.py

app.py

@@ -1,55 +1,44 @@
-import openai
 import streamlit as st
-from tenacity import retry, stop_after_attempt, wait_fixed
-
-# Set the OpenAI API key
-openai.api_key = st.secrets["YOUR_OPENAI_API_KEY"]
-
-initial_messages = [{
-    "role": "system", 
-    "content": """You are an AI assistant that provides highly customized, detailed, and unique lead generation strategies 
-for businesses. In this role you should envision yourself as a marketing innovator. You have theorized and tested countless unique marketing 
-strategies and built a reputation for your innovative ideas. You'll receive information about the type of business, the kind of customer 
-they'd like to attract, and any current assets they may have. 
-After receiving this information, follow these steps:
-1. Consider a wide range of 50 initial lead generation strategy ideas. Make sure you consider at least 10 fringe ideas. The kinds of lead gen strategies you wouldn't find
-in articles on the internet. These should be designed to fit any unique details the user gives you or simply very out of the box types of strategies. 
-2. Narrow it down to the top five best strategies based on the type of business, target customer, and current assets, including at least 2 very unique approaches.
-3. Generate detailed paragraphs for each of the top five strategies. Make sure the fourth and fifth strategies are completely unique and original.
-4. For each strategy, write a full paragraph suggestions the steps they should follow to implement that approach. 
-5. Range the strategies from most affordable to most expensive.
-6. Reply to the user with the detailed list of recommended strategies. """
-}]
-
-@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 CustomMarketingStrategy(business_description, customer_profile, messages):
-    messages.append({
-        "role": "user", 
-        "content": f"My business is about {business_description}. My ideal customer is {customer_profile}."
-    })
-    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.set_page_config(layout="wide")
-st.title("Marketing Strategy Suggester")
+import openai
 
-business_description = st.text_area("Describe Your Business", placeholder="Provide a brief description of your business, including the industry and main offerings.")
-customer_profile = st.text_area("Describe Your Ideal Customer", placeholder="Describe your ideal customer, including demographics, interests, and any other relevant details.")
+# Assuming your OpenAI API key is set correctly in your Hugging Face Secrets
+openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-submit_button = st.button('Get Marketing Strategies')
+# 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."""
+    }
+]
 
-if submit_button:
+# Function to handle the chat interaction
+def custom_chat_gpt(home_details, location):
     messages = initial_messages.copy()
-    reply, _ = CustomMarketingStrategy(business_description, customer_profile, messages)
-    st.write(reply)
-else:
-    st.write("**Submit your business and customer descriptions to receive marketing strategy suggestions.**")
+    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?"
+    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
+
+# 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')
+
+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.")