Update app.py

app.py

@@ -1,32 +1,50 @@
-import streamlit as st
 import openai
+import streamlit as st
+from tenacity import retry, stop_after_attempt, wait_fixed
 
-# Set up the OpenAI API key from Streamlit's secrets
-openai.api_key = st.secrets["OPENAI_API_KEY"]
+# Streamlit app initialization
+st.title("AI-powered Home Staging Assistant")
 
-st.title("Website Feature Code Generator")
+# If using Streamlit Cloud or Hugging Face Spaces, set up your OPENAI_API_KEY in the secrets.toml file
+# Otherwise, replace "your_actual_openai_api_key_here" with your actual OpenAI API key
+openai.api_key = st.secrets.get("OPENAI_API_KEY", "your_actual_openai_api_key_here")
 
-# User inputs for the feature they want to add to their website
-feature_description = st.text_area("Describe the website feature you need code for", placeholder="e.g., A contact form with name, email, and message fields")
+# Define the initial system message for the AI
+initial_messages_staging = [{
+    "role": "system",
+    "content": """
+        You are an AI assistant that provides home staging recommendations for real estate agents. Your role is to help agents visualize and execute staging setups that highlight the best features of the homes they're selling. When providing recommendations, ensure:
+        1. They are actionable and specific to the home's features and the target buyer demographic.
+        2. You include low-cost, DIY options for agents working with limited budgets.
+        3. You suggest staging strategies that can make the space appear larger and more inviting.
+        4. You consider current home decor trends and advise on how they can be applied effectively.
+        5. You offer ideas for both indoor and outdoor spaces.
+        Always conclude your response with a suggestion for professional staging services for agents who prefer expert assistance.
+    """
+}]
 
-if st.button('Generate Code'):
-    # Construct the prompt for code generation
-    prompt_text = f"Write HTML and CSS code for a website feature described as follows: {feature_description}."
+# Add retry logic to the OpenAI API call
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(1))
+def call_openai_api(messages):
+    return openai.ChatCompletion.create(
+        model="gpt-4",
+        messages=messages
+    )
 
-    # Call the OpenAI API for code generation
-    try:
-        response = openai.Completion.create(
-            model="code-davinci-002",  # Use a model suitable for code generation
-            prompt=prompt_text,
-            max_tokens=150  # Adjust max tokens as necessary
-        )
-        generated_code = response.choices[0].text
-    except Exception as e:
-        generated_code = f"Error in generating code: {e}"
+# Function to generate the AI response
+def generate_staging_advice(home_details, location):
+    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 = initial_messages_staging.copy()
+    messages.append({"role": "user", "content": user_input})
+    response = call_openai_api(messages)
+    return response.choices[0].message["content"]
 
-    # Display the generated code
-    st.subheader("Generated Code")
-    st.code(generated_code, language='html')
+# Streamlit UI elements for user input
+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')
 
-# Disclaimer for users
-st.write("Disclaimer: This tool provides AI-generated code suggestions. The generated code may require further testing and modification to be used in a production environment.")
+# Handling the button click to generate staging advice
+if generate_button:
+    staging_advice = generate_staging_advice(home_details, location)
+    st.text_area("Staging Recommendations", staging_advice, height=300)