Update app.py

app.py

@@ -11,32 +11,31 @@ repo = "artificialguybr/TshirtDesignRedmond-V2"
 trigger_word = "T shirt design, TshirtDesignAF, "
 
 # Function to generate image based on the prompt
-def generate_image(a, color_prompt, dress_type_prompt, design_prompt, prompt_part5, typography, text, contrast, shadows):
-    prompt_parts = [
-        a,
-        color_prompt,
-        dress_type_prompt,
-        design_prompt,
-        prompt_part5 == "hangs effortlessly on a plain grey wall, its simplicity transformed by bold",
-        shadows == "Soft light casts dynamic shadows, adding depth and emphasizing the crisp lines of the design, evoking a sense of modern sophistication."
-    ]
-
-    # Optional parts
+def infer(color_prompt, dress_type_prompt, design_prompt, text):
+    # Build the full prompt
+    prompt = (
+        f"a single {color_prompt} colored {dress_type_prompt} {design_prompt} "
+        "hangs effortlessly on a wall, its simplicity transformed by bold"
+    )
+    
+    # Conditional parts
     if text:
-        typography = "contemporary typography reading"
-        contrast = "The contrast between the text and the calm background creates a striking visual"
-        prompt_parts.extend([typography, text, contrast])
-
-    # Combine all parts into a full prompt
-    prompt = " ".join([part for part in prompt_parts if part])
+        prompt += (
+            f" contemporary typography reading {text} "
+            "The contrast between the text and the calm background creates a striking visual"
+        )
+    
+    # Add the hidden shadows part
+    prompt += f"Soft light casts dynamic shadows, adding depth and emphasizing the crisp lines of the design, evoking a sense of modern sophistication."
+    
+    full_prompt = f"{prompt} {trigger_word}"
 
-    print("Generating image with prompt:", prompt)
+    print("Generating image with prompt:", full_prompt)
     api_url = f"https://api-inference.huggingface.co/models/{repo}"
     #token = os.getenv("API_TOKEN")  # Uncomment and use your Hugging Face API token
     headers = {
         #"Authorization": f"Bearer {token}"
     }
-    full_prompt = f"{prompt} {trigger_word}"
     payload = {
         "inputs": full_prompt,
         "parameters": {
@@ -67,23 +66,18 @@ def generate_image(a, color_prompt, dress_type_prompt, design_prompt, prompt_par
 
 # Gradio Interface
 iface = gr.Interface(
-    fn=generate_image,
+    fn=infer,
     inputs=[
-        gr.Textbox(visible=False, placeholder="Hidden Part 1"),  # a (hidden)
         gr.Textbox(lines=1, placeholder="Color Prompt"),         # color_prompt
         gr.Textbox(lines=1, placeholder="Dress Type Prompt"),    # dress_type_prompt
         gr.Textbox(lines=2, placeholder="Design Prompt"),        # design_prompt
         gr.Textbox(lines=1, placeholder="Text (Optional)"),      # text
-        gr.Textbox(visible=False, placeholder="Hidden Part 5"),  # Hangs effortlessly (hidden)
-        gr.Textbox(visible=False, placeholder="Hidden Part 6"),  # Typography (hidden)
-        gr.Textbox(visible=False, placeholder="Hidden Part 8"),  # Contrast (hidden)
-        gr.Textbox(visible=False, placeholder="Hidden Part 9"),  # Shadows (hidden)
     ],
     outputs="image",
-    title="Clothe Designs to use in our img2img model",
-    description="Make designs for your clothes",
-    examples=[["a part", "Red", "T-shirt", "U-tube"]]
+    title="Clothe Designs for img2img model",
+    description="Generate designs for clothing items",
+    examples=[["a part", "Red", "T-shirt", "Simple design", "Stylish Text", "Soft Shadows"]]
 )
 
 print("Launching Gradio interface...")
-iface.launch()
+iface.launch()