From 2a480981414518e791d5e99d172a7b64ede19ce9 Mon Sep 17 00:00:00 2001
From: Kaustubh Maske Patil <37668193+nikochiko@users.noreply.github.com>
Date: Fri, 20 Oct 2023 18:20:20 +0530
Subject: [PATCH] Add breadcrumbs

---
 daras_ai_v2/base.py    | 28 +++++++++++++++++++++++-----
 gooey_ui/components.py | 16 ++++++++++++++++
 2 files changed, 39 insertions(+), 5 deletions(-)

diff --git a/daras_ai_v2/base.py b/daras_ai_v2/base.py
index f5c485c37..9442b8b46 100644
--- a/daras_ai_v2/base.py
+++ b/daras_ai_v2/base.py
@@ -26,6 +26,7 @@
 import gooey_ui as st
 from app_users.models import AppUser, AppUserTransaction
 from bots.models import SavedRun, Workflow
+from daras_ai.image_input import truncate_text_words
 from daras_ai_v2 import settings
 from daras_ai_v2.api_examples_widget import api_example_generator
 from daras_ai_v2.copy_to_clipboard_button_widget import (
@@ -146,6 +147,14 @@ def api_url(self, example_id=None, run_id=None, uid=None) -> furl:
     def endpoint(self) -> str:
         return f"/v2/{self.slug_versions[0]}/"
 
+    def get_page_title(self) -> str | None:
+        if (page_title := st.session_state.get(StateKeys.page_title)) != self.title:
+            return page_title
+        elif (text_prompt := st.session_state.get("text_prompt")):
+            return truncate_text_words(text_prompt, maxlen=60)
+        else:
+            return self.title
+
     def render(self):
         with sentry_sdk.configure_scope() as scope:
             scope.set_extra("base_url", self.app_url())
@@ -175,11 +184,20 @@ def render(self):
             StateKeys.page_notes, self.preview_description(st.session_state)
         )
 
-        root_url = self.app_url(example_id=example_id)
-        st.write(
-            f'# <a style="text-decoration: none;" target="_top" href="{root_url}">{st.session_state.get(StateKeys.page_title)}</a>',
-            unsafe_allow_html=True,
-        )
+        if example_id or run_id:
+            with st.breadcrumbs(className="mt-5"):
+                st.breadcrumb_item(
+                    self.title.upper(),
+                    link_to=self.app_url(),
+                    className="text-muted",
+                    style={"background-color": "#A5FFEE"}
+                )
+            st.write(f"# {self.get_page_title()}")
+        else:
+            with st.link(to=self.app_url(), className="text-decoration-none", target="_blank"):
+                st.write(f"# {self.title}")
+
+
         st.write(st.session_state.get(StateKeys.page_notes))
 
         try:
diff --git a/gooey_ui/components.py b/gooey_ui/components.py
index 4c19343e9..b8bb9596a 100644
--- a/gooey_ui/components.py
+++ b/gooey_ui/components.py
@@ -727,6 +727,22 @@ def _input_widget(
     return value
 
 
+def breadcrumbs(divider: str = "/", **props) -> state.NestingCtx:
+    style = props.pop("style", {}) | {"--bs-breadcrumb-divider": f"'{divider}'"}
+    with tag("nav", style=style, **props):
+        return tag("ol", className="breadcrumb mb-0")
+
+
+def breadcrumb_item(inner_html: str, link_to: str | None = None, **props):
+    className = "breadcrumb-item lead " + props.pop("className", "")
+    with tag("li", className=className, **props):
+        if link_to:
+            with tag("a", href=link_to):
+                html(inner_html)
+        else:
+            html(inner_html)
+
+
 def dedent(text: str | None) -> str | None:
     if not text:
         return text