Update app.py

app.py

@@ -646,8 +646,8 @@ def main():
                         view_options = ['Show as Markdown', 'Show as Code Editor', 'Show as Edit, Save, Run AI', 'Clone Document', 'New Record']
                         selected_view = st.sidebar.selectbox("Select Viewer/Editor", view_options, index=2)
                         
-                        Label = '# 📄 Markdown view - Mark it down, mark it up'
                         if selected_view == 'Show as Markdown':
+                            Label = '# 📄 Markdown view - Mark it down, mark it up'
                             st.markdown(Label)
                             total_docs = len(documents)
                             doc = documents[st.session_state.current_index]
@@ -691,8 +691,8 @@ def main():
 
 
                         
-                        Label = '# 💻 Code editor view - Code is poetry, bugs are typos'
                         elif selected_view == 'Show as Code Editor':
+                            Label = '# 💻 Code editor view - Code is poetry, bugs are typos'
                             st.markdown(Label)
                             total_docs = len(documents)
                             doc = documents[st.session_state.current_index]
@@ -735,8 +735,8 @@ def main():
 
 
                                     
-                        Label = '# ✏️ Edit and save view - Edit with wisdom, save with precision'
                         elif selected_view == 'Show as Edit, Save, Run AI':
+                            Label = '# ✏️ Edit and save view - Edit with wisdom, save with precision'
                             st.markdown(Label)
                             st.markdown("#### Edit the document fields below:")
                             
@@ -792,8 +792,8 @@ def main():
 
                                                         
 
-                        Label = '# 🧬 Clone functionality - Copy wisely, paste precisely'
                         elif selected_view == 'Clone Document':
+                            Label = '# 🧬 Clone functionality - Copy wisely, paste precisely'
                             st.markdown(Label)
                             st.markdown("#### Clone a document:")
                             for idx, doc in enumerate(documents_to_display):
@@ -830,8 +830,8 @@ def main():
 
 
                                             
-                            Label = '# 🆕 New Record view - Every new record is a new beginning'
                             elif selected_view == 'New Record':
+                                Label = '# 🆕 New Record view - Every new record is a new beginning'
                                 st.markdown(Label)
                                 st.markdown("#### Create a new document:")
                                 #if st.button("🤖 Insert Auto-Generated Record"):
@@ -867,10 +867,10 @@ def main():
 
 
             
-            Label = '# 📊 Data display - Data tells tales that words cannot'
             st.subheader(f"📊 Container: {st.session_state.selected_container}")
             if st.session_state.selected_container:
                 if documents_to_display:
+                    Label = '# 📊 Data display - Data tells tales that words cannot'
                     st.markdown(Label)
                     df = pd.DataFrame(documents_to_display)
                     st.dataframe(df)
@@ -936,11 +936,11 @@ def main():
 
 
             
-            Label = '# 💬 Chat functionality - Every chat is a chance to learn'
             st.subheader("💬 Chat with Claude")
             user_input = st.text_area("Message 📨:", height=100)
 
             if st.button("Send 📨"):
+                Label = '# 💬 Chat functionality - Every chat is a chance to learn'
                 st.markdown(Label)
                 if user_input:
                     response = client.messages.create(
@@ -1024,8 +1024,8 @@ def main():
 
 
     
-    Label = '# 🚪 Logout - All good things must come to an end'
     if st.session_state.logged_in and st.sidebar.button("🚪 Logout"):
+        Label = '# 🚪 Logout - All good things must come to an end'
         st.markdown(Label)
         st.session_state.logged_in = False
         st.session_state.selected_records.clear()