Update app.py

app.py

@@ -3,7 +3,7 @@ import streamlit as st
 from tenacity import retry, stop_after_attempt, wait_fixed
 
 # Assuming the OpenAI API key is stored in Streamlit's secrets
-openai.api_key = st.secrets["YOUR_OPENAI_API_KEY"]
+openai.api_key = st.secrets["OPENAI_API_KEY"]
 
 st.title("Website Feature Code Generator")
 
@@ -14,14 +14,16 @@ feature_description = st.text_area(
 )
 
 @retry(stop=stop_after_attempt(3), wait=wait_fixed(1))
-def call_openai_api(prompt):
+def call_openai_api(feature_description):
     try:
-        response = openai.Completion.create(
+        response = openai.ChatCompletion.create(
             model="gpt-4",
-            prompt=prompt,
-            max_tokens=150
+            messages=[
+                {"role": "system", "content": "You are a helpful AI trained in web development."},
+                {"role": "user", "content": feature_description}
+            ],
         )
-        return response.choices[0].text
+        return response.choices[0].message['content']
     except Exception as e:
         st.error(f"An error occurred: {str(e)}")
         return None