flyteorg · Future-Outlier · Oct 1, 2024 · Oct 1, 2024 · Oct 1, 2024 · Oct 2, 2024
@@ -121,6 +121,11 @@ def _dispatch_execute(
         input_proto = utils.load_proto_from_file(_literals_pb2.LiteralMap, local_inputs_file)
         idl_input_literals = _literal_models.LiteralMap.from_flyte_idl(input_proto)
 
+        # # TEST DECK
+        # if task_def is not None and not getattr(task_def, "disable_deck", True):
+        #     _output_deck(task_def.name.split(".")[-1], ctx.user_space_params)
+        # print("@@@ OUTPUT DECK @@@")
+
         # Step2
         # Invoke task - dispatch_execute
         outputs = task_def.dispatch_execute(ctx, idl_input_literals)

diff --git a/flytekit/deck/__init__.py b/flytekit/deck/__init__.py
@@ -18,5 +18,5 @@
    SourceCodeRenderer
 """
 
-from .deck import Deck, DeckField
+from .deck import Deck, DeckField, DummyDeck
 from .renderer import MarkdownRenderer, SourceCodeRenderer, TopFrameRenderer
@@ -11,6 +11,40 @@
 
 OUTPUT_DIR_JUPYTER_PREFIX = "jupyter"
 DECK_FILE_NAME = "deck.html"
+DUMMY_DECK_HTML = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <meta http-equiv="refresh" content="5" >
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <meta http-equiv="refresh" content="5" >
+    <title>Flytekit Status</title>
+    <style>
+        body {
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            height: 100vh;
+            margin: 0;
+            font-family: Arial, sans-serif;
+            background-color: #f0f0f0;
+        }
+        .message {
+            background-color: #ffffff;
+            padding: 20px;
+            border: 1px solid #cccccc;
+            border-radius: 5px;
+            box-shadow: 0 0 10px rgba(0, 0, 0, 0.1);
+            text-align: center;
+        }
+    </style>
+</head>
+<body>
+    <div class="message">
+        <p> Flyte decks have not been created yet. </p>
+    </div>
+</body>
+</html>
+"""
 
 
 class DeckField(str, enum.Enum):
@@ -86,6 +120,27 @@ def name(self) -> str:
     def html(self) -> str:
         return self._html
 
+    @classmethod
+    def publish(cls):
+        task_name = FlyteContextManager.current_context().user_space_params.task_id.name
+        new_user_params = FlyteContextManager.current_context().user_space_params
+        _output_deck(task_name, new_user_params)
+
+
+class DummyDeck(Deck):
+    """
+    The DummyDeck class is designed
+    """
+
+    def __init__(self):
+        name = "dummy_deck"
+        html = DUMMY_DECK_HTML
+        super().__init__(name, html)
+
+    @property
+    def html(self) -> str:
+        return self._html
+
 
 class TimeLineDeck(Deck):
     """
@@ -154,7 +209,13 @@ def _get_deck(
     Get flyte deck html string
     If ignore_jupyter is set to True, then it will return a str even in a jupyter environment.
     """
-    deck_map = {deck.name: deck.html for deck in new_user_params.decks}
+
+    deck_map = {deck.name: deck.html for deck in new_user_params.decks if deck.name != "dummy_deck"}
+
+    # If deck_map is empty after filtering, add DummyDeck
+    if not deck_map:
+        deck_map = {"dummy_deck": DUMMY_DECK_HTML}
+
     nav_htmls = []
     body_htmls = []