Update app.py

app.py

@@ -1,54 +1,42 @@
 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"]
 
-st.title("Holiday Video Backdrop Image Generator")
+st.title("Customer Avatar Generator")
 
-# User inputs for image generation
-image_theme = st.text_input("Image Theme", placeholder="Enter a theme for your holiday video backdrop (e.g., festive living room, snowy landscape)")
-color_scheme = st.text_input("Color Scheme", placeholder="Preferred color scheme (e.g., red and green, white and blue)")
-additional_elements = st.text_input("Additional Elements", placeholder="Any specific elements to include in the image (e.g., Christmas tree, snowflakes)")
+# 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)")
 
-# Dropdown for selecting image dimensions
-video_format = st.selectbox("Video Format", ["16:9 (Standard HD - 1920x1080)", "16:9 (4K - 3840x2160)", "1:1 (Square - 1080x1080)", "4:5 (Portrait - 1080x1350)", "9:16 (Tall - 1080x1920)"])
+submit_button = st.button("Generate Customer Avatar")
 
-# Mapping formats to DALL-E API size parameters
-format_size_mapping = {
-    "16:9 (Standard HD - 1920x1080)": "1024x576",  # Closest available size in DALL-E for 1080p
-    "16:9 (4K - 3840x2160)": "1792x1024",  # Closest available size in DALL-E for 4K
-    "1:1 (Square - 1080x1080)": "1024x1024",
-    "4:5 (Portrait - 1080x1350)": "1024x1792",  # Closest available vertical size
-    "9:16 (Tall - 1080x1920)": "1024x1792"  # Closest available vertical size
-}
+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}
+    """
 
-if st.button('Generate Image'):
-    # Construct the prompt
-    prompt = f"A holiday-themed backdrop image depicting: '{image_theme}', color scheme: '{color_scheme}', including elements: '{additional_elements}'."
-
-    # Call the DALL-E API
+    # Call the OpenAI API
     try:
-        response = openai.Image.create(
-            model="dall-e-3",  # Specify DALL-E 3 model
+        response = openai.Completion.create(
+            model="gpt-4",  # Specify GPT-4 model
             prompt=prompt,
-            n=1,
-            size=format_size_mapping[video_format],  # Set image dimensions based on selected video format
-            quality="hd"  # Set image quality to high definition
+            max_tokens=500,  # Adjust based on the required length of the avatar
         )
 
-        # Assuming the response contains a URL to the image
-        image_url = response['data'][0]['url']
-        
-        # Fetch the image from the URL
-        image_response = requests.get(image_url)
-        image = Image.open(BytesIO(image_response.content))
+        avatar = response.choices[0].text.strip()
 
-        # Display the image
-        st.image(image, caption='Generated Holiday Backdrop Image')
+        # Display the customer avatar
+        st.subheader("Generated Customer Avatar")
+        st.write(avatar)
 
     except Exception as e:
         st.error(f"An error occurred: {e}")