Skip to content

Commit

Permalink
Ask for confirmation before removing a chat
Browse files Browse the repository at this point in the history
  • Loading branch information
paulovcmedeiros committed Nov 13, 2023
1 parent c84833f commit 29df587
Showing 1 changed file with 15 additions and 1 deletion.
16 changes: 15 additions & 1 deletion pyrobbot/app/multipage.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ def add_page(self, page: AppPage, selected: bool = True, **page_obj_kwargs):
if selected:
self.register_selected_page(page)

def remove_page(self, page: AppPage):
def _remove_page(self, page: AppPage):
"""Remove a page from the app."""
self.pages[page.page_id].chat_obj.private_mode = True
self.pages[page.page_id].chat_obj.clear_cache()
Expand All @@ -66,6 +66,20 @@ def remove_page(self, page: AppPage):
except StopIteration:
self.add_page()

def remove_page(self, page: AppPage):
"""Remove a page from the app after confirmation."""
st.error("Are you sure you want to delete this chat?")
col1, col2 = st.columns([0.5, 0.5])
with col1:
st.button("No, take me back", use_container_width=True)
with col2:
st.button(
"Yes, delete chat",
on_click=self._remove_page,
kwargs={"page": page},
use_container_width=True,
)

def register_selected_page(self, page: AppPage):
"""Register a page as selected."""
self.state["selected_page"] = page
Expand Down

0 comments on commit 29df587

Please sign in to comment.