Added new column 'deploymeny configs' to the itable that copies the config code into user's clipboard

Author: Mohamed Zeidan

examples/inference/SDK/jumpstart_public_hub_visualization_utils.py

@@ -19,6 +19,7 @@
 import itables
 import pandas
 import logging
+import json
 from botocore.config import Config
 from ipywidgets import Button, Output
 from IPython.display import display
@@ -160,6 +161,7 @@ def _get_model_summary(self, full_summary):
             "Model Type": model_type,
             "Model Description": full_summary["HubContentDescription"],
             "Search Keywords": keywords,
+            "Deployment Configs": self._create_config_link(full_summary["HubContentName"]),
         }
 
     def _determine_model_type(self, keywords, model_id):
@@ -180,6 +182,63 @@ def _get_hub_document(self, model_id):
             HubContentType="Model", 
             HubContentName=model_id
         )["HubContentDocument"]
+    
+    def _get_supported_instance_types(self, model_id):
+        """Extract supported instance types from hub document."""
+        try:
+            hub_doc = self._get_hub_document(model_id)
+            doc_data = json.loads(hub_doc)
+            
+            supported_types = doc_data.get("SupportedInferenceInstanceTypes", ["ml.g5.8xlarge"])
+            default_type = doc_data.get("DefaultInferenceInstanceType")
+            
+            if default_type and default_type in supported_types:
+                supported_types = [default_type] + [t for t in supported_types if t != default_type]
+            
+            return supported_types
+        except Exception:
+            return ["ml.g5.8xlarge"]
+    
+    def _create_config_link(self, model_id):
+        """Create an HTML link that generates deployment config."""
+        config_code = self._generate_deployment_config(model_id).replace('\n', '\\n').replace("'", "\\'")
+        return f'<a href="javascript:void(0)" onclick="navigator.clipboard.writeText(\'{config_code}\'); alert(\'Config copied to clipboard!\')" style="color: #007bff; text-decoration: underline;">Get Config</a>'
+    
+    def _generate_deployment_config(self, model_id):
+        """Generate deployment configuration code for a model."""
+        supported_types = self._get_supported_instance_types(model_id)
+        default_type = supported_types[0] if supported_types else "ml.g5.8xlarge"
+        
+        types_comment = f"# Supported instance types: {', '.join(supported_types)}"
+        
+        config_code = f'''# Deployment configuration for {model_id}
+from sagemaker.hyperpod.inference.config.hp_jumpstart_endpoint_config import Model, Server, SageMakerEndpoint, TlsConfig
+from sagemaker.hyperpod.inference.hp_jumpstart_endpoint import HPJumpStartEndpoint
+
+{types_comment}
+
+# Create configs
+model = Model(
+    model_id='{model_id}',
+    model_version='<specify-version>',  # Use latest version or specify version
+)
+server = Server(
+    instance_type='{default_type}',  # Choose from supported types above
+)
+endpoint_name = SageMakerEndpoint(name='<your-endpoint-name>')
+tls_config = TlsConfig(tls_certificate_output_s3_uri='s3://<your-tls-bucket>')
+
+# Create endpoint spec
+js_endpoint = HPJumpStartEndpoint(
+    model=model,
+    server=server,
+    sage_maker_endpoint=endpoint_name,
+    tls_config=tls_config,
+)
+
+# Deploy the endpoint
+js_endpoint.create()'''
+        return config_code
 
 
 def get_all_public_hub_model_data(region: str):
@@ -198,7 +257,7 @@ def interactive_view(tabular_data: list):
     styled_df = _style_dataframe(df)
     layout = _get_table_layout(len(tabular_data))
 
-    itables.show(styled_df, layout=layout)
+    itables.show(styled_df, layout=layout, allow_html=True)
 
 
 def _configure_itables():