import openai

# Initialize OpenAI API
openai.api_key = "your_openai_api_key_here"

# Outfit suggestions database
outfit_database = {
    "casual": {
        "red jacket": ["white t-shirt", "blue jeans", "white sneakers", "black crossbody bag"],
        "denim skirt": ["striped blouse", "tan sandals", "straw hat", "neutral tote bag"]
    },
    "formal": {
        "red jacket": ["black turtleneck", "black trousers", "pointed heels", "gold necklace"],
        "denim skirt": ["silk blouse", "nude pumps", "pearl earrings", "clutch bag"]
    },
    # Add more clothing items and styles here
}

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

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."

def create_outfit_image(prompt):
    # Generate an image using OpenAI's DALL-E API
    response = openai.Image.create(
        prompt=prompt,
        n=1,
        size="1024x1024"
    )
    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)