Update app.py

app.py

@@ -1,50 +1,32 @@
-import openai
 import streamlit as st
-from tenacity import retry, stop_after_attempt, wait_fixed
+import openai
 
-# Assuming your OpenAI API key is set in Hugging Face Spaces secrets
-openai.api_key = st.secrets["YOUR_OPENAI_API_KEY"]
+# Set up the OpenAI API key from Streamlit's secrets
+openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-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."""
-    }
-]
+st.title("Website Feature Code Generator")
 
-@retry(stop=stop_after_attempt(3), wait=wait_fixed(1))
-def call_openai_api(messages):
-    return openai.ChatCompletion.create(
-        model="gpt-4",  # Make sure to use the correct model you have access to
-        messages=messages
-    )
+# 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")
 
-def CustomChatGPT(user_input, messages):
-    messages.append({"role": "user", "content": user_input})
-    response = call_openai_api(messages)
-    ChatGPT_reply = response.choices[0].message["content"]
-    messages.append({"role": "assistant", "content": ChatGPT_reply})
-    return ChatGPT_reply, messages
+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}."
 
-def wrapped_chat_gpt(home_details, location):
-    messages = initial_messages_staging.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?"
-    reply, updated_messages = CustomChatGPT(user_input, messages)
-    return reply
+    # 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}"
 
-# Streamlit Interface
-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')
+    # Display the generated code
+    st.subheader("Generated Code")
+    st.code(generated_code, language='html')
 
-if generate_button:
-    reply = wrapped_chat_gpt(home_details, location)
-    st.text(reply)
+# 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.")