Create fashion_tool.py

fashion_tool.py

@@ -0,0 +1,53 @@
+import gradio as gr
+from PIL import Image
+from diffusers import StableDiffusionPipeline
+import torch
+
+# Load the Stable Diffusion pipeline
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5").to(device)
+
+def generate_design(mood_board, style_preference, complexity, color_tone):
+    # Construct the prompt for the AI model
+    prompt = f"A {style_preference} fashion design with {color_tone} tones and complexity level {complexity}. Inspired by high fashion trends."
+    
+    # Add details from the mood board if provided
+    if mood_board:
+        prompt += " This design also integrates themes from the uploaded mood board."
+
+    # Generate the image using Stable Diffusion
+    with torch.autocast(device):
+        image = pipe(prompt).images[0]  # Generate the image
+
+    # Generate a text description
+    description = f"Design Style: {style_preference}, Tone: {color_tone}, Complexity: {complexity}"
+    return image, description
+
+# Create inputs and outputs
+mood_board_input = gr.Image(type="pil", label="Upload Mood Board (optional)")
+style_preference_input = gr.Dropdown(
+    choices=["Casual", "Formal", "Sporty", "Bohemian", "Avant-garde"],
+    label="Style Preference"
+)
+complexity_input = gr.Slider(1, 10, step=1, label="Complexity Level")
+color_tone_input = gr.Radio(
+    choices=["Bright", "Muted", "Neutral", "Pastel"],
+    label="Color Tone"
+)
+
+outputs = [
+    gr.Image(label="Generated Fashion Design"),
+    gr.Textbox(label="Design Details")
+]
+
+# Create the Gradio interface
+interface = gr.Interface(
+    fn=generate_design,
+    inputs=[mood_board_input, style_preference_input, complexity_input, color_tone_input],
+    outputs=outputs,
+    title="AI-Assisted Designer Tool",
+    description="Generate unique fashion designs based on mood boards, style preferences, and other inputs."
+)
+
+# Launch the app
+interface.launch()