Update app.py

app.py

@@ -221,7 +221,6 @@ def enhance_image(
     randomize_seed,
     num_inference_steps,
     upscale_factor,
-    guidance_scale,
     denoising_strength,
     use_generated_caption,
     custom_prompt,
@@ -272,7 +271,7 @@ def enhance_image(
         control_image,
         denoising_strength,
         num_inference_steps,
-        guidance_scale,
+        1.0,  # Hardcoded guidance_scale to 1
         generator,
         tile_size=1024,
         overlap=32
@@ -286,11 +285,11 @@ def enhance_image(
 
 
 # Create Gradio interface
-with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as demo:
+with gr.Blocks(css=css, title="🎨 AI Image Upscaler - Florence-2 + FLUX") as demo:
     gr.HTML("""
     <div class="main-header">
-        <h1>🎨 AI Image Enhancer</h1>
-        <p>Upload an image or provide a URL to enhance it using Florence-2 captioning and FLUX upscaling</p>
+        <h1>🎨 AI Image Upscaler</h1>
+        <p>Upload an image or provide a URL to upscale it using Florence-2 captioning and FLUX upscaling</p>
         <p>Currently running on <strong>{}</strong></p>
     </div>
     """.format(power_device))
@@ -304,7 +303,7 @@ with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as d
                     input_image = gr.Image(
                         label="Upload Image",
                         type="pil",
-                        height=300
+                        height=200  # Made smaller
                     )
                 
                 with gr.TabItem("🔗 Image URL"):
@@ -328,7 +327,7 @@ with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as d
                 lines=2
             )
             
-            gr.HTML("<h3>⚙️ Enhancement Settings</h3>")
+            gr.HTML("<h3>⚙️ Upscaling Settings</h3>")
             
             upscale_factor = gr.Slider(
                 label="Upscale Factor",
@@ -348,15 +347,6 @@ with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as d
                 info="More steps = better quality but slower"
             )
             
-            guidance_scale = gr.Slider(
-                label="Guidance Scale",
-                minimum=1.0,
-                maximum=10.0,
-                step=0.5,
-                value=3.5,
-                info="How closely to follow the prompt"
-            )
-            
             denoising_strength = gr.Slider(
                 label="Denoising Strength",
                 minimum=0.0,
@@ -381,19 +371,20 @@ with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as d
                 )
             
             enhance_btn = gr.Button(
-                "🚀 Enhance Image",
+                "🚀 Upscale Image",
                 variant="primary",
                 size="lg"
             )
 
-        with gr.Column(scale=1):
+        with gr.Column(scale=2):  # Larger scale for results
             gr.HTML("<h3>📊 Results</h3>")
             
             result_slider = ImageSlider(
-                label="Input / Enhanced",
+                label="Input / Upscaled",
                 type="pil",
                 interactive=True,
-                height=400
+                height=600,  # Made larger
+                elem_id="result_slider"
             )
             
             with gr.Row():
@@ -410,26 +401,6 @@ with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as d
                 interactive=False
             )
 
-    # Examples
-    gr.Examples(
-        examples=[
-            [None, "https://upload.wikimedia.org/wikipedia/commons/thumb/a/a7/Example.jpg/800px-Example.jpg", 42, False, 25, 2, 3.5, 0.3, True, ""],
-            [None, "https://picsum.photos/512/512", 123, False, 25, 3, 4.0, 0.3, True, ""],
-        ],
-        inputs=[
-            input_image,
-            image_url,
-            seed,
-            randomize_seed,
-            num_inference_steps,
-            upscale_factor,
-            guidance_scale,
-            denoising_strength,
-            use_generated_caption,
-            custom_prompt,
-        ]
-    )
-
     # Event handler
     enhance_btn.click(
         fn=enhance_image,
@@ -440,7 +411,6 @@ with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as d
             randomize_seed,
             num_inference_steps,
             upscale_factor,
-            guidance_scale,
             denoising_strength,
             use_generated_caption,
             custom_prompt,
@@ -454,11 +424,26 @@ with gr.Blocks(css=css, title="🎨 AI Image Enhancer - Florence-2 + FLUX") as d
         <ol>
             <li><strong>Florence-2</strong> analyzes your image and generates a detailed caption</li>
             <li>Initial upscale with LANCZOS interpolation (or ESRGAN if installed)</li>
-            <li><strong>FLUX Img2Img</strong> enhances the upscaled image with tiled AI diffusion guided by the caption</li>
+            <li><strong>FLUX Img2Img</strong> upscales the upscaled image with tiled AI diffusion guided by the caption</li>
         </ol>
         <p><strong>Note:</strong> Output limited to 8192x8192 pixels total budget. Tiling enables larger sizes.</p>
     </div>
     """)
+    
+    # Custom CSS for slider
+    gr.HTML("""
+    <style>
+        #result_slider .slider {
+            width: 100% !important;
+            max-width: inherit !important;
+        }
+        #result_slider img {
+            object-fit: contain !important;
+            width: 100% !important;
+            height: auto !important;
+        }
+    </style>
+    """)
 
 if __name__ == "__main__":
     demo.queue().launch(share=True, server_name="0.0.0.0", server_port=7860)