Update app.py

app.py

@@ -11,16 +11,21 @@ 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, text, shadows):
-    prompt_parts = [a, color_prompt, dress_type_prompt, design_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]
 
     # Optional 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])
+        prompt_parts.extend([part_6, text, part_8])
 
-    prompt_parts.append(shadows)
+    prompt_parts.append(part_9)
 
     # Combine all parts into a full prompt
     prompt = " ".join([part for part in prompt_parts if part])
@@ -60,22 +65,19 @@ def generate_image(a, color_prompt, dress_type_prompt, design_prompt, text, shad
             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)"),                 # text
-        gr.Textbox(visible=False, placeholder="Hidden Part 9"),  # Shadows (hidden)
+        gr.Textbox(lines=1, placeholder="Text (Optional)", label="Optional Text"),  # text (optional)
     ],
     outputs="image",
-    title="Rookus Xpress",
+    title="Clothe Designs to use in our img2img model",
     description="Make designs for your clothes",
-    examples=[["a part", "Red", "T-shirt", "Simple design", "Stylish Text", "Soft Shadows"]]
+    examples=[["Red", "T-shirt", "Simple design", "Stylish Text"]]
 )
 
 print("Launching Gradio interface...")
-iface.launch()
+iface.launch()