Add MCP server

app.py

@@ -86,6 +86,15 @@ def first_image_from_dir(directory):
 
 # Function to reset the UI and state
 def reset_all():
+    """
+    Reset all UI components to their initial state when a new image is uploaded.
+    
+    This function is triggered when the user uploads a new image to clear all previous
+    results and prepare the interface for a new reconstruction session.
+    
+    Returns:
+        tuple: A tuple containing None values for all image components and reset UI states
+    """
     return (
         None,  # crop_img
         None,  # normals_img
@@ -227,11 +236,11 @@ def step4_track(session_id):
 # New: run all steps sequentially
 @spaces.GPU(duration=120)
 def generate_results_and_mesh(image, session_id=None):
-
     """
     Process an input image through a 3D reconstruction pipeline and return the intermediate outputs and mesh file.
 
-    This function runs a multi‐step workflow to go from a raw input image to a reconstructed 3D mesh:
+    This function is triggered when the user clicks the "Reconstruct Face" button or selects an example.
+    It runs a multi‐step workflow to go from a raw input image to a reconstructed 3D mesh:
       1. **Preprocessing**: crops and masks the image for object isolation.
       2. **Normals Estimation**: computes surface normal maps.
       3. **UV Mapping**: generates UV coordinate maps for texturing.
@@ -240,7 +249,7 @@ def generate_results_and_mesh(image, session_id=None):
 
     Args:
         image (PIL.Image.Image or ndarray): Input image to reconstruct.
-        session_id (str): Unique identifier for this session’s output directories.
+        session_id (str): Unique identifier for this session's output directories.
 
     Returns:
         tuple:
@@ -273,6 +282,16 @@ def generate_results_and_mesh(image, session_id=None):
 
 # Cleanup on unload
 def cleanup(request: gr.Request):
+    """
+    Clean up session-specific directories and temporary files when the user session ends.
+    
+    This function is triggered when the Gradio demo is unloaded (e.g., when the user
+    closes the browser tab or navigates away). It removes all temporary files and
+    directories created during the user's session to free up storage space.
+    
+    Args:
+        request (gr.Request): Gradio request object containing session information
+    """
     sid = request.session_hash
     if sid:
         d1 = os.path.join(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], sid)
@@ -281,6 +300,19 @@ def cleanup(request: gr.Request):
         shutil.rmtree(d2, ignore_errors=True)
 
 def start_session(request: gr.Request):
+    """
+    Initialize a new user session and return the session identifier.
+    
+    This function is triggered when the Gradio demo loads and creates a unique
+    session hash that will be used to organize outputs and temporary files
+    for this specific user session.
+    
+    Args:
+        request (gr.Request): Gradio request object containing session information
+        
+    Returns:
+        str: Unique session hash identifier
+    """
     return request.session_hash
 
     
@@ -350,4 +382,4 @@ with gr.Blocks(css=css) as demo:
 
 demo.queue()
 
-demo.launch(share=True)
+demo.launch(share=True, mcp_server=True)