Update app.py

app.py

@@ -11,13 +11,15 @@ 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, typography, text, contrast, shadows):
+def generate_image(a, color_prompt, dress_type_prompt, design_prompt, text, shadows):
     prompt_parts = [a, color_prompt, dress_type_prompt, design_prompt]
-    
+
     # 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.append(shadows)
 
     # Combine all parts into a full prompt
@@ -47,7 +49,7 @@ def generate_image(a, color_prompt, dress_type_prompt, design_prompt, typography
         print("API response status code:", response.status_code)
         if response.status_code == 200:
             print("Image generation successful!")
-            return Image.open(BytesIO(response.content))  # Changed to match the first code
+            return Image.open(BytesIO(response.content))
         elif response.status_code == 503:
             time.sleep(1)
             pbar.update(1)
@@ -66,15 +68,13 @@ iface = gr.Interface(
         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(visible=False, placeholder="Hidden Part 5"),  # Typography (hidden)
-        gr.Textbox(lines=1, optional=True, placeholder="Text"),  # text
-        gr.Textbox(visible=False, placeholder="Hidden Part 6"),  # Contrast (hidden)
+        gr.Textbox(lines=1, placeholder="Text"),                 # text
         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", "Simple design", "Bold Typography", "Stylish Text", "High Contrast", "Soft Shadows"]]
+    examples=[["a part", "Red", "T-shirt", "Simple design", "Stylish Text", "Soft Shadows"]]
 )
 
 print("Launching Gradio interface...")