Update app.py

app.py

@@ -1,42 +1,38 @@
 import streamlit as st
 import openai
+import requests
+from PIL import Image
+from io import BytesIO
 
-# Set up the OpenAI API key from Hugging Face Spaces secrets
-openai.api_key = st.secrets["OPENAI_API_KEY"]
+# Setup OpenAI API key (ensure it's set in environment variables)
+openai.api_key = os.environ["OPENAI_API_KEY"]
 
 st.title("Customer Avatar Generator")
 
-# User inputs for generating the customer avatar
-business_type = st.text_input("Business Type", "Enter the type of your business (e.g., Real Estate, E-commerce)")
-target_market = st.text_input("Target Market", "Describe your target market (e.g., first-time home buyers, fitness enthusiasts)")
-unique_selling_points = st.text_area("Unique Selling Points", "What sets your business apart from competitors? (e.g., personalized service, affordable pricing)")
-customer_challenges = st.text_area("Customer Challenges", "What challenges do your customers face that your business solves? (e.g., finding affordable housing, achieving fitness goals)")
-
-submit_button = st.button("Generate Customer Avatar")
-
-if submit_button:
-    # Construct the prompt for the AI
-    prompt = f"""
-    Generate a detailed customer avatar for a business.
-    Business Type: {business_type}
-    Target Market: {target_market}
-    Unique Selling Points: {unique_selling_points}
-    Customer Challenges: {customer_challenges}
-    """
-
-    # Call the OpenAI API
+# 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}"
     try:
-        response = openai.Completion.create(
-            model="gpt-4",  # Specify GPT-4 model
-            prompt=prompt,
-            max_tokens=500,  # Adjust based on the required length of the avatar
-        )
-
-        avatar = response.choices[0].text.strip()
-
-        # Display the customer avatar
-        st.subheader("Generated Customer Avatar")
-        st.write(avatar)
-
+        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')
     except Exception as e:
-        st.error(f"An error occurred: {e}")
+        st.error(f"Error in generating image: {e}")