update

app.py

@@ -1,7 +1,6 @@
 import json
 import gradio as gr
 from mcp_mlflow_tools import (
-    initialize_tracking,
     set_tracking_uri,
     get_system_info,
     list_experiments,
@@ -18,17 +17,6 @@ def create_interface():
         gr.Markdown("A service that exposes MLflow functionality through a web interface and API endpoints.")
         
         with gr.Tab("Tracking & System Info"):
-            with gr.Group():
-                gr.Markdown("## Initialize Tracking")
-                init_output = gr.JSON(label="Initialization Status")
-                init_button = gr.Button("Initialize / Check Tracking URI")
-                init_button.click(
-                    fn=initialize_tracking,
-                    inputs=[],
-                    outputs=init_output,
-                    api_name="initialize_tracking"
-                )
-            
             with gr.Group():
                 gr.Markdown("## Set Tracking URI")
                 uri_input = gr.Textbox(label="MLflow Tracking URI")

mcp_mlflow_tools.py

@@ -8,30 +8,6 @@ from mlflow.tracking import MlflowClient
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-def initialize_tracking(default_uri: str = "http://127.0.0.1:5000") -> Dict:
-    """
-    Checks if the MLflow tracking URI is set to a remote server.
-    If not, it sets it to a default and returns a notification message.
-    """
-    try:
-        current_uri = mlflow.get_tracking_uri()
-        # Default local URI is 'file:///.../mlruns'. We check if it's not a remote server.
-        if current_uri and (current_uri.startswith("http://") or current_uri.startswith("https://")):
-            message = f"MLflow tracking URI is configured to: {current_uri}. You can use the interface below to change it."
-            status = "already_set"
-        else:
-            logger.info(f"Tracking URI is not set to a remote server. Defaulting to {default_uri}")
-            mlflow.set_tracking_uri(default_uri)
-            message = f"Warning: MLflow tracking URI was not set. It has been automatically set to the default: {default_uri}. You can change this below."
-            status = "defaulted"
-        
-        return {"status": status, "message": message, "uri": mlflow.get_tracking_uri()}
-
-    except Exception as e:
-        error_msg = f"Failed during tracking initialization: {str(e)}"
-        logger.error(error_msg, exc_info=True)
-        return {"error": True, "message": error_msg}
-
 def _format_timestamp(ts: int) -> str:
     """Convert MLflow timestamp (milliseconds since epoch) to readable string."""
     dt = datetime.fromtimestamp(ts / 1000.0)