Add Python Dependency Deck

flytekit/core/context_manager.py

@@ -741,6 +741,17 @@ def get_deck(self) -> typing.Union[str, "IPython.core.display.HTML"]:  # type:ig
 
         return _get_deck(typing.cast(ExecutionState, self.execution_state).user_space_params)
 
+    def add_deck(self, deck: Deck) -> None:
+        """
+        Adds a deck to the current context.
+        """
+        if (
+            self.execution_state is not None
+            and self.execution_state.user_space_params is not None
+            and hasattr(self.execution_state.user_space_params, "decks")
+        ):
+            self.execution_state.user_space_params.decks.append(deck)
+
     @dataclass
     class Builder(object):
         file_access: FileAccessProvider

flytekit/deck/deck.py

@@ -174,3 +174,29 @@
         autoescape=select_autoescape(enabled_extensions=("html",)),
     )
     return env.get_template("template.html")
+
+
+class PythonDependencyDeck(Deck):
+    """
+    PythonDependencyDeck is a deck that contains information about the library name and version.
+    """
+
+    def __init__(self, name: Optional[str] = "Python Dependency", html: Optional[str] = ""):
+        super().__init__(name, html)
+
+    @property
+    def html(self) -> str:
+        import json
+        import subprocess
+
+        import pandas as pd
+        from flytekitplugins.deck.renderer import TableRenderer
+
+        try:
+            installed_packages = json.loads(subprocess.check_output(["pip", "list", "--format", "json"]))
+        except subprocess.CalledProcessError as e:
+            logger.error(f"Error occurred while fetching installed packages: {e}")
+            return ""
+        df = pd.DataFrame(installed_packages)
+        html = TableRenderer().to_html(df)
+        return html

tests/flytekit/unit/deck/test_deck.py

@@ -8,7 +8,7 @@
 import flytekit
 from flytekit import Deck, FlyteContextManager, task
 from flytekit.deck import MarkdownRenderer, SourceCodeRenderer, TopFrameRenderer
-from flytekit.deck.deck import _output_deck
+from flytekit.deck.deck import PythonDependencyDeck, _output_deck
 
 
 @pytest.mark.skipif("pandas" not in sys.modules, reason="Pandas is not installed.")
@@ -176,3 +176,12 @@ def test_source_code_renderer():
     # Assert that the color #ffffff is used instead of #fff0f0
     assert "#ffffff" in result
     assert "#fff0f0" not in result
+
+
+def test_python_dependency_deck():
+    python_dependency_deck = PythonDependencyDeck()
+    ctx = FlyteContextManager.current_context()
+    ctx.user_space_params.decks.clear()
+    ctx.add_deck(python_dependency_deck)
+    assert len(ctx.user_space_params.decks) == 1
+    assert ctx.user_space_params.decks[0].name == "Python Dependency"