Update

app.py

@@ -305,7 +305,6 @@ class HunyuanImageApp:
     def refine_image(self, 
                     image: Image.Image,
                     prompt: str,
-                    negative_prompt: str,
                     width: int,
                     height: int,
                     num_inference_steps: int,
@@ -315,6 +314,9 @@ class HunyuanImageApp:
         try:
             if image is None:
                 return None, "Please upload an image to refine."
+            
+            if not prompt or prompt.strip() == "":
+                return None, "Please enter a refinement prompt."
 
             torch.cuda.empty_cache()
 
@@ -332,7 +334,6 @@ class HunyuanImageApp:
             refined_image = self.pipeline.refiner_pipeline(
                 image=image,
                 prompt=prompt,
-                negative_prompt=negative_prompt,
                 width=width,
                 height=height,
                 num_inference_steps=num_inference_steps,
@@ -555,14 +556,8 @@ def create_interface(auto_load: bool = True, use_distilled: bool = False, device
                         
                         refine_prompt = gr.Textbox(
                             label="Refinement Prompt",
-                            placeholder="Make the image more detailed and high quality",
-                            lines=2,
-                            value="Make the image more detailed and high quality"
-                        )
-                        
-                        refine_negative_prompt = gr.Textbox(
-                            label="Negative Prompt",
-                            placeholder="",
+                            placeholder="Image description",
+                            info="This prompt should describe the image content.",
                             lines=2,
                             value=""
                         )
@@ -599,6 +594,8 @@ def create_interface(auto_load: bool = True, use_distilled: bool = False, device
                         refined_image = gr.Image(
                             label="Refined Image",
                             type="pil",
+                            format="png",
+                            show_download_button=True,
                             height=600
                         )
                         refinement_status = gr.Textbox(
@@ -633,7 +630,7 @@ def create_interface(auto_load: bool = True, use_distilled: bool = False, device
         refine_btn.click(
            fn=app.refine_image,
            inputs=[
-               input_image, refine_prompt, refine_negative_prompt,
+               input_image, refine_prompt,
                refine_width, refine_height, refine_steps, refine_guidance, refine_seed
            ],
            outputs=[refined_image, refinement_status]
@@ -685,7 +682,6 @@ if __name__ == "__main__":
     demo = create_interface(auto_load=auto_load, use_distilled=args.use_distilled, device=args.device)
     
     print("🚀 Starting HunyuanImage Gradio App...")
-    print(f"📱 The app will be available at: http://{args.host}:{args.port}")
     print(f"🔧 Auto-load pipeline: {'Yes' if auto_load else 'No'}")
     print(f"🎯 Model type: {'Distilled' if args.use_distilled else 'Non-distilled'}")
     print(f"💻 Device: {args.device}")