Update app.py

app.py

@@ -7,22 +7,28 @@ from io import BytesIO
 # Access the OpenAI API key from Hugging Face Spaces secrets
 openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-st.title("Customer Avatar Generator")
+st.title("Intuitive Customer Avatar Generator")
 
 # User inputs for avatar generation
+st.subheader("Tell Us About Your Business")
 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")
+primary_service_or_product = st.text_input("Primary Service/Product", placeholder="e.g., Home sales, Personal training sessions")
+unique_selling_points = st.text_input("Unique Selling Points", placeholder="What sets your business apart?")
+business_location = st.text_input("Business Location", placeholder="City or region you primarily serve")
+
+st.subheader("Your Customer Interactions")
+most_common_customer_feedback = st.text_area("Common Customer Feedback", placeholder="What feedback do you often receive from customers?")
+customer_challenges = st.text_input("Customer Challenges", placeholder="What challenges do your customers typically face?")
+repeat_customer_characteristics = st.text_input("Repeat Customer Characteristics", placeholder="Any common traits among your repeat customers?")
 
 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}."
+    prompt_text = f"Create a customer avatar based on the following business details: Business type: {business_type}, primary service/product: {primary_service_or_product}, unique selling points: {unique_selling_points}, business location: {business_location}. Customer feedback: {most_common_customer_feedback}, customer challenges: {customer_challenges}, repeat customer characteristics: {repeat_customer_characteristics}."
 
     # Call the OpenAI API for text generation
     try:
         response_text = openai.ChatCompletion.create(
-            model="gpt-4",  # Use GPT-4 model
+            model="gpt-4",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant."},
                 {"role": "user", "content": prompt_text}
@@ -32,28 +38,6 @@ if st.button('Generate Avatar'):
     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:
-        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
+    # Display the avatar description
     st.write("**Avatar Description:**")
     st.write(avatar_description)
-    if image:
-        st.image(image, caption='Generated Customer Avatar Image')