show model URL after uploading

Author: Borda

src/litmodels/io/cloud.py

@@ -2,7 +2,7 @@
 # Licensed under the Apache License, Version 2.0 (the "License");
 #     http://www.apache.org/licenses/LICENSE-2.0
 #
-from typing import Optional, Tuple
+from typing import Optional, Tuple, Union
 
 from lightning_sdk.api.teamspace_api import UploadedModelInfo
 from lightning_sdk.teamspace import Teamspace
@@ -15,6 +15,8 @@
 # else:
 #     LightningModule = None
 
+_SHOWED_MODEL_LINKS = []
+
 
 def _parse_name(name: str) -> Tuple[str, str, str]:
     """Parse the name argument into its components."""
@@ -50,11 +52,34 @@ def _get_teamspace(name: str, organization: str) -> Teamspace:
     return Teamspace(**teamspaces[requested_teamspace])
 
 
+def _print_model_link(org_name: str, teamspace_name: str, model_name: str, verbose: Union[bool, int]) -> None:
+    """Print a link to the uploaded model.
+
+    Args:
+        org_name: Name of the organization.
+        teamspace_name: Name of the teamspace.
+        model_name: Name of the model.
+        verbose: Whether to print the link:
+
+            - If set to 0, no link will be printed.
+            - If set to 1, the link will be printed only once.
+            - If set to 2, the link will be printed every time.
+    """
+    url = f"https://lightning.ai/{org_name}/{teamspace_name}/models/{model_name}"
+    msg = f"Model uploaded successfully. Link to the model: '{url}'"
+    if int(verbose) > 1:
+        print(msg)
+    elif url not in _SHOWED_MODEL_LINKS:
+        print(msg)
+        _SHOWED_MODEL_LINKS.append(url)
+
+
 def upload_model_files(
     name: str,
     path: str,
     progress_bar: bool = True,
     cluster_id: Optional[str] = None,
+    verbose: Union[bool, int] = 0,
 ) -> UploadedModelInfo:
     """Upload a local checkpoint file to the model store.
 
@@ -65,16 +90,20 @@ def upload_model_files(
         progress_bar: Whether to show a progress bar for the upload.
         cluster_id: The name of the cluster to use. Only required if it can't be determined
             automatically.
+        verbose: Whether to print a link to the uploaded model. If set to 0, no link will be printed.
 
     """
     org_name, teamspace_name, model_name = _parse_name(name)
     teamspace = _get_teamspace(name=teamspace_name, organization=org_name)
-    return teamspace.upload_model(
+    info = teamspace.upload_model(
         path=path,
         name=model_name,
         progress_bar=progress_bar,
         cluster_id=cluster_id,
     )
+    if verbose:
+        _print_model_link(org_name, teamspace_name, model_name, verbose)
+    return info
 
 
 def download_model_files(