Update app.py

app.py

@@ -11,21 +11,22 @@ repo = "artificialguybr/TshirtDesignRedmond-V2"
 trigger_word = "T shirt design, TshirtDesignAF, "
 
 # Function to generate image based on the prompt
-def generate_image(color_prompt, dress_type_prompt, design_prompt, text):
-    # Constructing the prompt with the specified parts
-    a = "a"
-    part_5 = "hangs effortlessly on a plain grey wall, its simplicity transformed by bold"
-    part_6 = "contemporary typography reading"
-    part_8 = "The contrast between the text and the calm background creates a striking visual"
-    part_9 = "Soft light casts dynamic shadows, adding depth and emphasizing the crisp lines of the design, evoking a sense of modern sophistication."
-    
-    prompt_parts = [a, color_prompt, dress_type_prompt, design_prompt, part_5]
+def generate_image(a, color_prompt, dress_type_prompt, design_prompt, text, shadows):
+    prompt_parts = [
+        a,
+        color_prompt,
+        dress_type_prompt,
+        design_prompt,
+        "hangs effortlessly on a plain grey wall, its simplicity transformed by bold",
+    ]
 
     # Optional parts
     if text:
-        prompt_parts.extend([part_6, text, part_8])
+        typography = "contemporary typography reading"
+        contrast = "The contrast between the text and the calm background creates a striking visual"
+        prompt_parts.extend([typography, text, contrast])
 
-    prompt_parts.append(part_9)
+    prompt_parts.append(shadows)
 
     # Combine all parts into a full prompt
     prompt = " ".join([part for part in prompt_parts if part])
@@ -65,19 +66,25 @@ def generate_image(color_prompt, dress_type_prompt, design_prompt, text):
             print("API Error:", response.status_code)
             raise Exception(f"API Error: {response.status_code}")
 
+# Gradio Interface
 iface = gr.Interface(
     fn=generate_image,
     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)", label="Optional Text"),  # text (optional)
+        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=[["Red", "T-shirt", "Simple design", "Stylish Text"]]
+    examples=[["a part", "Red", "T-shirt", "Simple design", "Stylish Text"]]
 )
 
 print("Launching Gradio interface...")
-iface.launch()
+iface.launch()