Update app.py

app.py

@@ -97,8 +97,6 @@ def add_extra_model_paths() -> None:
 add_comfyui_directory_to_sys_path()
 add_extra_model_paths()
 
-
-
 def import_custom_nodes() -> None:
     import asyncio
     import execution
@@ -195,8 +193,9 @@ getimagesizeandcount = NODE_CLASS_MAPPINGS["GetImageSizeAndCount"]()
 depthanything_v2 = NODE_CLASS_MAPPINGS["DepthAnything_V2"]()
 imageresize = NODE_CLASS_MAPPINGS["ImageResize+"]()
 
+
 @spaces.GPU
-def generate_image(prompt, structure_image, style_image, depth_strength=15, style_strength=0.5, progress=gr.Progress(track_tqdm=True)) -> str:
+def generate_image(structure_image, style_image, depth_strength=15, style_strength=0.5, progress=gr.Progress(track_tqdm=True)) -> str:
     """Main generation function that processes inputs and returns the path to the generated image."""
     with torch.inference_mode():
         # Set up CLIP
@@ -206,9 +205,9 @@ def generate_image(prompt, structure_image, style_image, depth_strength=15, styl
             clip2=get_value_at_index(CLIP_MODEL, 0),
         )
         
-        # Encode text
+        # Encode text with default prompt
         text_encoded = cliptextencode.encode(
-            text=prompt,
+            text="person wearing fashionable clothing",
             clip=get_value_at_index(clip_switch, 0),
         )
         empty_text = cliptextencode.encode(
@@ -328,17 +327,14 @@ def generate_image(prompt, structure_image, style_image, depth_strength=15, styl
         saved_path = f"output/{saved['ui']['images'][0]['filename']}"
         return saved_path
 
-
-
-output_image = gr.Image(label="Virtual Try-On Result")
-
-
-
 # Create Gradio interface
 examples = [
-    ["person wearing fashionable clothing", "f1.webp", "f11.webp", 15, 0.6],
-    ["person wearing elegant dress", "f2.webp", "f21.webp", 15, 0.5],
-    ["person wearing casual outfit", "f3.webp", "f31.webp", 15, 0.5],
+    ["f1.webp", "f11.webp", 15, 0.6],
+    ["f2.webp", "f21.webp", 15, 0.5],
+    ["f3.webp", "f31.webp", 15, 0.5],
+    ["qq1.webp", "ww1.webp", 15, 0.5],
+    ["qq2.webp", "ww2.webp", 15, 0.5],
+    ["qq3.webp", "ww3.webp", 15, 0.5]
 ]
 
 # Gradio 인터페이스 생성
@@ -353,10 +349,6 @@ with demo:
         with gr.TabItem("👔 Virtual Try-On"):
             with gr.Row():
                 with gr.Column():
-                    prompt_input = gr.Textbox(
-                        label="Style Description", 
-                        placeholder="Describe the desired style (e.g., 'person wearing elegant dress')"
-                    )
                     with gr.Row():
                         with gr.Group():
                             structure_image = gr.Image(
@@ -390,7 +382,7 @@ with demo:
             
             gr.Examples(
                 examples=examples,
-                inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
+                inputs=[structure_image, style_image, depth_strength, style_strength],
                 outputs=output_image,
                 fn=generate_image,
                 cache_examples=False
@@ -399,9 +391,9 @@ with demo:
             # Connect the button to the generation function
             generate_button.click(
                 fn=generate_image,
-                inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
+                inputs=[structure_image, style_image, depth_strength, style_strength],
                 outputs=output_image
             )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)