MCP enabled

app.py

@@ -180,6 +180,21 @@ def preprocess_image(input_image: Image.Image) -> Image.Image:
 
 @spaces.GPU()
 def preprocess_n_magnify(input_image: Image.Image):
+    """
+    Preprocess the input image and perform a single-step 4× magnification using the SeeSR pipeline.
+
+    This function first resizes the input to fit within a 512×512 thumbnail, then applies the full
+    magnification through ControlNet-guided diffusion—to produce a high-resolution, 4× upscaled image.
+
+    Args:
+        input_image (PIL.Image.Image): The source image to preprocess and magnify.
+
+    Returns:
+        tuple[PIL.Image.Image, PIL.Image.Image]:
+            - The resized (thumbnail) version of the input.
+            - The final 4× magnified output image.
+    """
+    
     processed_img = preprocess_image(input_image)
 
     img, magnified_img = magnify(processed_img)    
@@ -338,7 +353,7 @@ with gr.Blocks(css=css, theme=theme) as demo:
         inputs = [
             input_image,
         ]
-        run_button.click(fn=magnify, inputs=input_image, outputs=[result_gallery])
-        input_image.upload(fn=preprocess_image,inputs=input_image, outputs=input_image)
+        run_button.click(fn=preprocess_n_magnify, inputs=input_image, outputs=[result_gallery])
+        input_image.upload(fn=preprocess_image,inputs=input_image, outputs=input_image, show_api=False)
 
-demo.launch(share=True)
+demo.launch(share=True, mcp_server=True)