Add breadcrumb when run was taken from an example

GooeyAI · Nov 14, 2023 · aa8f334 · aa8f334
1 parent 40ff964
commit aa8f334
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/daras_ai_v2/base.py b/daras_ai_v2/base.py
@@ -196,13 +196,20 @@ def render(self):
         )
 
         if example_id or run_id:
-            with st.breadcrumbs(className="mt-5"):
+            with st.breadcrumbs(className="mt-5", divider="/"):
                 st.breadcrumb_item(
                     self.get_recipe_page_title(st.session_state).upper(),
                     link_to=self.app_url(),
                     className="text-muted",
-                    style={"background-color": "#A5FFEE"},
                 )
+                saved_run = self.get_sr_from_query_params(example_id, run_id, uid)
+                if saved_run.parent and saved_run.parent.example_id:
+                    st.breadcrumb_item(
+                        self.get_page_title(saved_run.parent.to_dict()),
+                        link_to=self.app_url(
+                            example_id=saved_run.parent.example_id,
+                        ),
+                    )
             st.write(f"# {self.get_page_title(st.session_state)}")
         else:
             with st.link(