Update app.py

app.py

@@ -7,26 +7,20 @@ 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("TikTok Video Backdrop and Image Generator for Small Businesses")
 
 # 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)")
+st.subheader("Define Your Business")
+business_type = st.text_input("Your Business Type", placeholder="e.g., Cafe, Florist, Yoga Studio")
 
-# Dropdown for selecting image dimensions
-video_format = st.selectbox("Video Format", ["16:9 (Standard HD - 1920x1080)", "1:1 (Square - 1080x1080)", "4:5 (Portrait - 1080x1350)"])
-
-# Mapping formats to DALL-E API size parameters
-format_size_mapping = {
-    "16:9 (Standard HD - 1920x1080)": "1024x576",  # Reduced size for 1080p
-    "1:1 (Square - 1080x1080)": "1024x1024",
-    "4:5 (Portrait - 1080x1350)": "1024x768",  # Reduced size for portrait
-}
+st.subheader("Video Backdrop Details")
+video_theme = st.text_input("Video Theme", placeholder="Enter a theme for your TikTok video (e.g., cozy cafe, floral arrangement, yoga session)")
+color_scheme = st.text_input("Color Scheme", placeholder="Preferred color scheme (e.g., pastel, vibrant, earth tones)")
+additional_elements = st.text_input("Additional Elements", placeholder="Any specific elements to include in the backdrop (e.g., coffee cups, flowers, yoga mats)")
 
 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}'."
+    prompt = f"Create a vertical TikTok video backdrop image for a {business_type}. Theme: '{video_theme}', color scheme: '{color_scheme}', including elements: '{additional_elements}'."
 
     # Call the DALL-E API
     try:
@@ -34,7 +28,7 @@ if st.button('Generate Image'):
             model="dall-e-3",  # Specify DALL-E 3 model
             prompt=prompt,
             n=1,
-            size=format_size_mapping[video_format],  # Set reduced image dimensions based on selected video format
+            size="1024x1792",  # Set image dimensions to vertical
             quality="standard"  # Set image quality to standard for lower cost
         )
 
@@ -46,7 +40,7 @@ if st.button('Generate Image'):
         image = Image.open(BytesIO(image_response.content))
 
         # Display the image
-        st.image(image, caption='Generated Holiday Backdrop Image')
+        st.image(image, caption='Generated Business Video Backdrop Image')
 
     except Exception as e:
         st.error(f"An error occurred: {e}")