feat: Enable MCP

app.py

@@ -105,7 +105,20 @@ def resize_and_crop_image(image):
     return cropped_image, message, original_size
 
 def handle_image_upload(image):
-    """Handle image upload and store original size globally"""
+    """
+    Handle image upload and preprocessing for the Gradio interface.
+    
+    This function is called when a user uploads an image to the real images tab.
+    It stores the original image size globally and processes the image to the required dimensions.
+    
+    Args:
+        image: PIL.Image object uploaded by the user, or None if no image is uploaded.
+    
+    Returns:
+        Tuple containing:
+        - processed_image: PIL.Image object resized and cropped to 1024x1024, or None if no image.
+        - message: HTML-formatted string indicating the processing status, or empty string.
+    """
     global original_image_size
     
     if image is None:
@@ -132,7 +145,30 @@ def process_generated_image(
     localization_model,
     progress=gr.Progress(track_tqdm=True)
 ):
-    """Process generated image with ADDIT"""
+    """
+    Process and generate images using ADDIT for the generated images workflow.
+    
+    This function generates a source image from a text prompt and then adds an object to it
+    based on the target prompt and subject token using the ADDIT pipeline.
+    
+    Args:
+        prompt_source: String describing the source scene without the object to be added.
+        prompt_target: String describing the target scene including the object to be added.
+        subject_token: String token representing the object to add (must appear in target prompt).
+        seed_src: Integer seed for generating the source image.
+        seed_obj: Integer seed for generating the object.
+        extended_scale: Float value (1.0-1.3) controlling the extended attention scale.
+        structure_transfer_step: Integer (0-10) controlling structure transfer strength.
+        blend_steps: String of comma-separated integers for blending steps, or empty string.
+        localization_model: String specifying the localization model to use.
+        progress: Gradio progress tracker for displaying progress updates.
+    
+    Returns:
+        Tuple containing:
+        - src_image: PIL.Image of the generated source image, or None if error.
+        - edited_image: PIL.Image with the added object, or None if error.
+        - status_message: String describing the result or error message.
+    """
     global pipe
     
     if pipe is None:
@@ -199,7 +235,33 @@ def process_real_image(
     disable_inversion,
     progress=gr.Progress(track_tqdm=True)
 ):
-    """Process real image with ADDIT"""
+    """
+    Process and edit a real uploaded image using ADDIT to add objects.
+    
+    This function takes an uploaded image and adds an object to it based on the target prompt
+    and subject token using the ADDIT pipeline with optional inversion and offset techniques.
+    
+    Args:
+        source_image: PIL.Image object of the uploaded source image to edit.
+        prompt_source: String describing the source image content.
+        prompt_target: String describing the desired result including the object to add.
+        subject_token: String token representing the object to add (must appear in target prompt).
+        seed_src: Integer seed for source image processing.
+        seed_obj: Integer seed for object generation.
+        extended_scale: Float value (1.0-1.3) controlling the extended attention scale.
+        structure_transfer_step: Integer (0-10) controlling structure transfer strength.
+        blend_steps: String of comma-separated integers for blending steps, or empty string.
+        localization_model: String specifying the localization model to use.
+        use_offset: Boolean indicating whether to use offset technique.
+        disable_inversion: Boolean indicating whether to disable DDIM inversion.
+        progress: Gradio progress tracker for displaying progress updates.
+    
+    Returns:
+        Tuple containing:
+        - src_image: PIL.Image of the processed source image, or None if error.
+        - edited_image: PIL.Image with the added object, or None if error.
+        - status_message: String describing the result or error message.
+    """
     global pipe
     
     if pipe is None:
@@ -506,5 +568,6 @@ demo = create_interface()
 demo.launch(
     server_name="0.0.0.0",
     server_port=7860,
-    share=True
-) 
+    share=True,
+    mcp_server=True
+)