Update app.py

app.py

@@ -1,6 +1,7 @@
 import openai
+import streamlit as st
 
-# Initialize OpenAI API
+# Set your OpenAI API key
 openai.api_key = "your_openai_api_key_here"
 
 # Outfit suggestions database
@@ -16,26 +17,25 @@ outfit_database = {
     # Add more clothing items and styles here
 }
 
+# Function to generate outfit advice
 def generate_outfit_advice(piece, color, style):
-    # Find the clothing piece in the database
     key = f"{color} {piece}" if f"{color} {piece}" in outfit_database.get(style, {}) else piece
     suggestions = outfit_database.get(style, {}).get(key, None)
     
     if not suggestions:
         return "Sorry, I couldn't find an outfit for your request. Try another combination!"
 
-    # Generate outfit advice
     top, bottom, footwear, accessory = suggestions
     advice = (f"Here’s how you can style your {color} {piece} for a {style} look:\n"
               f"- Top: {top}\n- Bottom: {bottom}\n- Footwear: {footwear}\n- Accessory: {accessory}")
     return advice
 
+# Function to generate image prompt for OpenAI's DALL-E
 def generate_image_prompt(piece, color, style):
-    # Create a text prompt for image generation
     return f"A {style} outfit featuring a {color} {piece} styled with complementary clothing items and accessories. Modern, fashionable, and cohesive."
 
+# Function to create outfit image using OpenAI's DALL-E
 def create_outfit_image(prompt):
-    # Generate an image using OpenAI's DALL-E API
     response = openai.Image.create(
         prompt=prompt,
         n=1,
@@ -43,18 +43,34 @@ def create_outfit_image(prompt):
     )
     return response["data"][0]["url"]
 
-# User inputs
-piece = input("Enter the clothing piece (e.g., 'jacket', 'skirt'): ").lower()
-color = input("Enter the color (e.g., 'red', 'black'): ").lower()
-style = input("Enter the style (e.g., 'casual', 'formal'): ").lower()
-
-# Generate outfit advice and image
-advice = generate_outfit_advice(piece, color, style)
-image_prompt = generate_image_prompt(piece, color, style)
-
-if "Sorry" not in advice:
-    image_url = create_outfit_image(image_prompt)
-    print(advice)
-    print(f"Generated Image: {image_url}")
-else:
-    print(advice)
+# Streamlit UI
+def main():
+    st.title("AI Fashion Outfit Generator")
+    st.write("Enter the details of your clothing piece, color, and style to get personalized fashion advice and a visual representation.")
+
+    # Form to collect user inputs
+    with st.form(key='outfit_form'):
+        piece = st.text_input("Enter the clothing piece (e.g., 'jacket', 'skirt')").lower()
+        color = st.text_input("Enter the color (e.g., 'red', 'black')").lower()
+        style = st.selectbox("Select style", ["casual", "formal", "streetwear", "boho"])
+
+        submit_button = st.form_submit_button(label='Generate Outfit')
+
+    if submit_button:
+        if piece and color:
+            # Generate advice and image based on user input
+            advice = generate_outfit_advice(piece, color, style)
+            image_prompt = generate_image_prompt(piece, color, style)
+            
+            if "Sorry" not in advice:
+                # Display the outfit advice and image
+                st.write(advice)
+                image_url = create_outfit_image(image_prompt)
+                st.image(image_url, caption="Generated Outfit", use_column_width=True)
+            else:
+                st.error(advice)
+        else:
+            st.warning("Please enter both a clothing piece and a color.")
+
+if __name__ == '__main__':
+    main()