feat: Enable MCP

app.py

@@ -82,17 +82,25 @@ def predict(
     chunk_size: int = 512,
 ):
     """
-    Predict the response for the input text and url.
+    Generate a response for the given issue description and GitHub repository.
+    
+    This function clones a GitHub repository, analyzes its structure, locates relevant files
+    based on the issue description, and generates a repair solution using the Kimi-Dev model.
+    
     Args:
-        text (str): The input text.
-        url (str): The input url.
-        chatbot (list): The chatbot.
-        history (list): The history.
-        top_p (float): The top-p value.
-        temperature (float): The temperature value.
-        repetition_penalty (float): The repetition penalty value.
-        max_length_tokens (int): The max length tokens.
-        chunk_size (int): The chunk size.
+        text: The issue description provided by the user
+        url: The GitHub repository URL
+        commit_hash: The specific commit hash to checkout
+        chatbot: The current chatbot conversation state
+        history: The conversation history
+        top_p: The top-p sampling parameter for text generation
+        temperature: The temperature parameter for text generation
+        max_length_tokens: Maximum number of tokens to generate
+        chunk_size: Size of chunks for streaming response (default: 512)
+        
+    Returns:
+        Generator yielding tuples of (chatbot_state, history_state, status_message) representing
+        the updated conversation state and generation status
     """
     print("running the prediction function")
 
@@ -201,7 +209,25 @@ def retry(
     chunk_size: int = 512,
 ):
     """
-    Retry the response for the input text and url.
+    Regenerate the response for the previous input.
+    
+    This function retries the prediction with the same parameters as the last request,
+    useful when the user wants to get a different response for the same input.
+    
+    Args:
+        text: The issue description from the previous request
+        url: The GitHub repository URL from the previous request
+        commit_hash: The commit hash from the previous request
+        chatbot: The current chatbot conversation state
+        history: The conversation history
+        top_p: The top-p sampling parameter for text generation
+        temperature: The temperature parameter for text generation
+        max_length_tokens: Maximum number of tokens to generate
+        chunk_size: Size of chunks for streaming response (default: 512)
+        
+    Returns:
+        Generator yielding tuples of (chatbot_state, history_state, status_message) or
+        yields error state if history is empty
     """
     if len(history) == 0:
         yield (chatbot, history, "Empty context")
@@ -258,6 +284,18 @@ def build_demo(args: argparse.Namespace) -> gr.Blocks:
                     retry_btn = gr.Button("🔄 Regenerate")
 
             def respond(message):
+                """
+                Handle the submission of URL and commit hash.
+                
+                This function provides feedback when the user submits a GitHub URL and commit hash,
+                confirming that the information has been received.
+                
+                Args:
+                    message: The input message (not used in current implementation)
+                    
+                Returns:
+                    A confirmation message string
+                """
                 return f"Url and commit hash submitted!"
             with gr.Column():
                 url_box = gr.Textbox(label="Please input a Github url here",placeholder="Input your url", lines=1)
@@ -328,7 +366,8 @@ def main(args: argparse.Namespace):
         favicon_path=favicon_path,
         server_name=args.ip,
         server_port=args.port,
-        share=True 
+        share=True,
+        mcp_server=True
     )
 
 if __name__ == "__main__":
@@ -341,4 +380,4 @@ if __name__ == "__main__":
 
     args = parse_args()
     print(args)
-    main(args)
+    main(args)