Update app.py

app.py

@@ -281,7 +281,10 @@ def enhance_image(
         gr.Info(f"📏 Resizing output to target size: {w_original * upscale_factor}x{h_original * upscale_factor}")
         image = image.resize((w_original * upscale_factor, h_original * upscale_factor), resample=Image.LANCZOS)
     
-    return [true_input_image, 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]
 
 
 # Create Gradio interface
@@ -383,7 +386,8 @@ with gr.Blocks(css=css, title="🎨 AI Image Upscaler - Florence-2 + FLUX") as d
                 type="pil",
                 interactive=True,
                 height=600,  # Made larger
-                elem_id="result_slider"
+                elem_id="result_slider",
+                label=None  # Remove default label
             )
 
     # Event handler
@@ -424,6 +428,37 @@ with gr.Blocks(css=css, title="🎨 AI Image Upscaler - Florence-2 + FLUX") as d
         #result_slider .gr-button-tool {
             display: none !important;
         }
+        #result_slider .gr-button-undo {
+            display: none !important;
+        }
+        #result_slider .gr-button-clear {
+            display: none !important;
+        }
+        #result_slider .badge-container .badge {
+            display: none !important;
+        }
+        #result_slider .badge-container::before {
+            content: "Before";
+            position: absolute;
+            top: 10px;
+            left: 10px;
+            background: rgba(0,0,0,0.5);
+            color: white;
+            padding: 5px;
+            border-radius: 5px;
+            z-index: 10;
+        }
+        #result_slider .badge-container::after {
+            content: "After";
+            position: absolute;
+            top: 10px;
+            right: 10px;
+            background: rgba(0,0,0,0.5);
+            color: white;
+            padding: 5px;
+            border-radius: 5px;
+            z-index: 10;
+        }
     </style>
     """)