Update app.py

app.py

@@ -1,26 +1,23 @@
 import gradio as gr
 import requests
-from PIL import Image, ImageOps, ImageChops
+import os
+from PIL import Image
 from io import BytesIO
 from tqdm import tqdm
 import time
 
-# Repository information and trigger word
+# Defining the repository information and the trigger word
 repo = "artificialguybr/TshirtDesignRedmond-V2"
-dress_trigger = "Minimalist dress design, DressDesignAF"
-design_trigger = "Clothing design pattern, DesignPatternAF"
+trigger_word = "T shirt design, TshirtDesignAF, "
 
-def generate_image(prompt, style="dress"):
-    """
-    Generate an image using Hugging Face API with a specific style.
-    """
-    print(f"Generating {style} image with prompt:", prompt)
+def generate_image(prompt):
+    print("Generating image with prompt:", prompt)
     api_url = f"https://api-inference.huggingface.co/models/{repo}"
-    # token = os.getenv("API_TOKEN")  # Uncomment and use your Hugging Face API token
+    #token = os.getenv("API_TOKEN")  # Uncomment and use your Hugging Face API token
     headers = {
-        # "Authorization": f"Bearer {token}"
+        #"Authorization": f"Bearer {token}"
     }
-    full_prompt = f"{prompt} {dress_trigger}" if style == "dress" else f"{prompt} {design_trigger}"
+    full_prompt = f"{prompt} {trigger_word}"
     payload = {
         "inputs": full_prompt,
         "parameters": {
@@ -38,7 +35,7 @@ def generate_image(prompt, style="dress"):
         print("API response status code:", response.status_code)
         if response.status_code == 200:
             print("Image generation successful!")
-            return Image.open(BytesIO(response.content))
+            return Image.open(BytesIO(response.content))  # Changed to match the first code
         elif response.status_code == 503:
             time.sleep(1)
             pbar.update(1)
@@ -49,67 +46,14 @@ def generate_image(prompt, style="dress"):
             print("API Error:", response.status_code)
             raise Exception(f"API Error: {response.status_code}")
 
-def blend_design(dress_img, design_img):
-    """
-    Blend the design image onto the dress image.
-    """
-    # Ensure both images are the same size
-    dress_img = ImageOps.fit(dress_img, (512, 512))
-    design_img = ImageOps.fit(design_img, (512, 512))
-    
-    # Create an alpha mask for blending
-    design_mask = design_img.convert("L").point(lambda x: min(255, x * 2))  # Enhance design visibility
-    blended = Image.composite(design_img, dress_img, design_mask)
-    
-    return blended
+iface = gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(lines=2, placeholder="Type your prompt here..."),
+    outputs="image",
+    title="Clothe Designs to use in our img2img model",
+    description="Make designs for your clothes",
+    examples=[["Cute Panda"], ["Skull"]]
+)
 
-# Define the interface for generating the minimalist dress
-def generate_minimalist_dress(dress_type_prompt, color_prompt):
-    """
-    Generate a minimalist dress based on type and color prompts.
-    """
-    prompt = f"{dress_type_prompt}, {color_prompt}"
-    return generate_image(prompt, style="dress")
-
-# Define the interface for generating the design
-def generate_clothing_design(design_prompt):
-    """
-    Generate a design image based on the provided design prompt.
-    """
-    return generate_image(design_prompt, style="design")
-
-# Define the interface for blending the dress and design
-def blend_images(dress_image, design_image):
-    """
-    Blend a dress image and a design image together.
-    """
-    return blend_design(dress_image, design_image)
-
-# Gradio interface
-iface = gr.Blocks()
-
-with iface:
-    gr.Markdown("# Minimalist Dress and Design Generator")
-    
-    with gr.Tab("Generate Minimalist Dress"):
-        dress_type = gr.Textbox(label="Dress Type Prompt", placeholder="e.g., A-line dress, sleeveless dress")
-        color = gr.Textbox(label="Color Prompt", placeholder="e.g., Red, pastel pink")
-        dress_output = gr.Image(label="Generated Dress")
-        dress_button = gr.Button("Generate Dress")
-        dress_button.click(generate_minimalist_dress, inputs=[dress_type, color], outputs=dress_output)
-    
-    with gr.Tab("Generate Design"):
-        design_prompt = gr.Textbox(label="Design Prompt", placeholder="e.g., Floral pattern, geometric design")
-        design_output = gr.Image(label="Generated Design")
-        design_button = gr.Button("Generate Design")
-        design_button.click(generate_clothing_design, inputs=[design_prompt], outputs=design_output)
-    
-    with gr.Tab("Blend Dress and Design"):
-        dress_input = gr.Image(type="pil", label="Upload Dress Image")
-        design_input = gr.Image(type="pil", label="Upload Design Image")
-        blended_output = gr.Image(label="Blended Output")
-        blend_button = gr.Button("Blend Images")
-        blend_button.click(blend_images, inputs=[dress_input, design_input], outputs=blended_output)
-
-# Launch interface
+print("Launching Gradio interface...")
 iface.launch()