Update app.py

app.py

@@ -4,35 +4,56 @@ import requests
 from PIL import Image
 from io import BytesIO
 
-# Setup OpenAI API key (ensure it's set in environment variables)
-openai.api_key = os.environ["OPENAI_API_KEY"]
+# Access the OpenAI API key from Hugging Face Spaces secrets
+openai.api_key = st.secrets["OPENAI_API_KEY"]
 
 st.title("Customer Avatar Generator")
 
 # User inputs for avatar generation
-name = st.text_input("Customer's Name", placeholder="Enter a name for your customer avatar")
-age = st.text_input("Age Range", placeholder="e.g., 30-40 years")
-gender = st.selectbox("Gender", ["Female", "Male", "Non-binary", "Prefer not to say"])
-location = st.text_input("Location", placeholder="e.g., Urban, Suburban, Rural")
-interests = st.text_area("Interests", placeholder="List their interests, hobbies, and preferences")
-profession = st.text_input("Profession", placeholder="e.g., Software Engineer, Teacher")
-income_level = st.text_input("Income Level", placeholder="e.g., $50,000 - $70,000 annually")
-
-# Button for generating avatar
-if st.button('Generate Customer Avatar'):
-    # Generating text profile
-    profile_prompt = f"Create a detailed customer avatar profile for: Name: {name}, Age: {age}, Gender: {gender}, Location: {location}, Interests: {interests}, Profession: {profession}, Income Level: {income_level}."
-    profile_response = openai.Completion.create(model="gpt-4", prompt=profile_prompt, max_tokens=150)
-    profile = profile_response.choices[0].text.strip()
-    st.subheader("Customer Avatar Profile:")
-    st.write(profile)
-
-    # Generating image
-    image_prompt = f"An image of a person who fits this profile: {profile_prompt}"
+business_type = st.text_input("Business Type", placeholder="e.g., Real Estate, Fitness")
+target_demographics = st.text_input("Target Demographics", placeholder="e.g., Age range, gender, location")
+interests = st.text_input("Interests", placeholder="e.g., Home improvement, health and wellness")
+additional_details = st.text_input("Additional Details", placeholder="Any specific characteristics or preferences")
+
+if st.button('Generate Avatar'):
+    # Construct the prompt for text generation
+    prompt_text = f"Create a detailed customer avatar for a business in the {business_type} industry. Target demographics: {target_demographics}. Interests: {interests}. Additional details: {additional_details}."
+
+    # Call the OpenAI API for text generation
+    try:
+        response_text = openai.ChatCompletion.create(
+            model="gpt-4",  # Use GPT-4 model
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": prompt_text}
+            ]
+        )
+        avatar_description = response_text.choices[0].message['content']
+    except Exception as e:
+        avatar_description = f"Error in generating avatar description: {e}"
+
+    # Construct the prompt for image generation
+    prompt_image = f"A visual representation of a customer avatar for a {business_type} business. Demographics: {target_demographics}. Interests: {interests}."
+
+    # Call the DALL-E API for image generation
     try:
-        image_response = openai.Image.create(model="dall-e", prompt=image_prompt, n=1)
-        image_url = image_response['data'][0]['url']
-        image = Image.open(BytesIO(requests.get(image_url).content))
-        st.image(image, caption='Generated Avatar Image')
+        response_image = openai.Image.create(
+            model="dall-e-3",  # Specify DALL-E 3 model
+            prompt=prompt_image,
+            n=1,
+            size="1024x1024"  # Set image dimensions
+        )
+
+        # Fetch the image from the URL
+        image_url = response_image['data'][0]['url']
+        image_response = requests.get(image_url)
+        image = Image.open(BytesIO(image_response.content))
     except Exception as e:
+        image = None
         st.error(f"Error in generating image: {e}")
+
+    # Display the results
+    st.write("**Avatar Description:**")
+    st.write(avatar_description)
+    if image:
+        st.image(image, caption='Generated Customer Avatar Image')