From d20794e271a4e3ef8f7f250b65d28b678e74adf3 Mon Sep 17 00:00:00 2001
From: LUIS NOVO <lfnovo@gmail.com>
Date: Wed, 13 Nov 2024 17:17:16 -0300
Subject: [PATCH] correct context options for sources and notes

---
 pages/stream_app/consts.py | 8 +++++++-
 pages/stream_app/note.py   | 4 ++--
 pages/stream_app/source.py | 4 ++--
 3 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/pages/stream_app/consts.py b/pages/stream_app/consts.py
index 352b338..eb2446a 100644
--- a/pages/stream_app/consts.py
+++ b/pages/stream_app/consts.py
@@ -1,4 +1,10 @@
-context_icons = [
+source_context_icons = [
+    "⛔ not in context",
+    "🟡 summary",
+    "🟢 full content",
+]
+
+note_context_icons = [
     "⛔ not in context",
     "🟡 summary",
     "🟢 full content",
diff --git a/pages/stream_app/note.py b/pages/stream_app/note.py
index 4a1c5be..8c85523 100644
--- a/pages/stream_app/note.py
+++ b/pages/stream_app/note.py
@@ -9,7 +9,7 @@ from open_notebook.graphs.multipattern import graph as pattern_graph
 from open_notebook.utils import surreal_clean
 from pages.components import note_panel
 
-from .consts import context_icons
+from .consts import note_context_icons
 
 
 @st.dialog("Write a Note", width="large")
@@ -60,7 +60,7 @@ def note_card(note, notebook_id):
         context_state = st.selectbox(
             "Context",
             label_visibility="collapsed",
-            options=context_icons,
+            options=note_context_icons,
             index=1,
             key=f"note_{note.id}",
         )
diff --git a/pages/stream_app/source.py b/pages/stream_app/source.py
index 1a2c358..13978b4 100644
--- a/pages/stream_app/source.py
+++ b/pages/stream_app/source.py
@@ -13,7 +13,7 @@ from open_notebook.exceptions import UnsupportedTypeException
 from open_notebook.graphs.source import source_graph
 from pages.components import source_panel
 
-from .consts import context_icons
+from .consts import source_context_icons
 
 
 @st.dialog("Source", width="large")
@@ -113,7 +113,7 @@ def source_card(source, notebook_id):
         context_state = st.selectbox(
             "Context",
             label_visibility="collapsed",
-            options=context_icons,
+            options=source_context_icons,
             index=1,
             key=f"source_{source.id}",
         )