Update app.py

app.py

@@ -12,6 +12,7 @@ from gradio_imageslider import ImageSlider
 from PIL import Image
 from huggingface_hub import snapshot_download
 import requests
+import io
 
 # For ESRGAN (requires pip install basicsr gfpgan)
 try:
@@ -233,7 +234,6 @@ def tiled_flux_img2img(pipe, prompt, image, strength, steps, guidance, generator
 def enhance_image(
     image_input,
     image_url,
-    seed,
     randomize_seed,
     num_inference_steps,
     upscale_factor,
@@ -251,8 +251,16 @@ def enhance_image(
     else:
         raise gr.Error("Please provide an image (upload or URL)")
 
+    # Convert input image to PNG in backend
+    buffer = io.BytesIO()
+    input_image.save(buffer, format="PNG")
+    buffer.seek(0)
+    input_image = Image.open(buffer)
+
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
+    else:
+        seed = 42
 
     true_input_image = input_image
     
@@ -300,15 +308,15 @@ def enhance_image(
     # Resize input image to match output size for slider alignment
     resized_input = true_input_image.resize(image.size, resample=Image.LANCZOS)
     
-    return [resized_input, image]
+    return [resized_input, image], image
 
 
 # Create Gradio interface
-with gr.Blocks(css=css, title="🎨 AI Image Upscaler - Florence-2 + FLUX") as demo:
+with gr.Blocks(css=css, title="🎨 Flux dev Creative Upscaler - Florence-2 + FLUX") as demo:
     gr.HTML("""
     <div class="main-header">
-        <h1>🎨 AI Image Upscaler</h1>
-        <p>Upload an image or provide a URL to upscale it using Florence-2 captioning and FLUX upscaling</p>
+        <h1>🎨 Flux dev Creative Upscaler</h1>
+        <p>Upload an image or provide a URL to upscale it using Florence-2 captioning and FLUX dev with Ultimate SD Upscaler</p>
         <p>Currently running on <strong>{}</strong></p>
     </div>
     """.format(power_device))
@@ -358,7 +366,7 @@ with gr.Blocks(css=css, title="🎨 AI Image Upscaler - Florence-2 + FLUX") as d
             )
             
             num_inference_steps = gr.Slider(
-                label="Number of Inference Steps",
+                label="Steps (25 Recommended)",
                 minimum=8,
                 maximum=50,
                 step=1,
@@ -367,7 +375,7 @@ with gr.Blocks(css=css, title="🎨 AI Image Upscaler - Florence-2 + FLUX") as d
             )
             
             denoising_strength = gr.Slider(
-                label="Denoising Strength",
+                label="Creativity (Denoising)",
                 minimum=0.0,
                 maximum=1.0,
                 step=0.05,
@@ -380,14 +388,6 @@ with gr.Blocks(css=css, title="🎨 AI Image Upscaler - Florence-2 + FLUX") as d
                     label="Randomize seed",
                     value=True
                 )
-                seed = gr.Slider(
-                    label="Seed",
-                    minimum=0,
-                    maximum=MAX_SEED,
-                    step=1,
-                    value=42,
-                    interactive=True
-                )
             
             enhance_btn = gr.Button(
                 "🚀 Upscale Image",
@@ -406,13 +406,20 @@ with gr.Blocks(css=css, title="🎨 AI Image Upscaler - Florence-2 + FLUX") as d
                 label=None  # Remove default label
             )
 
+            upscaled_output = gr.Image(
+                label="Upscaled Image (Download as PNG)",
+                type="pil",
+                interactive=False,
+                show_download_button=True,
+                height=600,
+            )
+
     # Event handler
     enhance_btn.click(
         fn=enhance_image,
         inputs=[
             input_image,
             image_url,
-            seed,
             randomize_seed,
             num_inference_steps,
             upscale_factor,
@@ -420,7 +427,7 @@ with gr.Blocks(css=css, title="🎨 AI Image Upscaler - Florence-2 + FLUX") as d
             use_generated_caption,
             custom_prompt,
         ],
-        outputs=[result_slider]
+        outputs=[result_slider, upscaled_output]
     )
     
     gr.HTML("""