escape markdown special chars

Signed-off-by: Gianluca Capuzzi <gianluca.posta78@gmail.com>

Author: gcapuzzi

.gitignore

@@ -23,12 +23,13 @@ vectordb.txt
 
 # Local dataset
 dataset/
+dataset*/
 
 # Editor e sistema
 .DS_Store
 .idea/
 .vscode/
 
 responses.txt
-vectordb.txt
-secrets.toml
+secrets.toml
+users.db

app.py

@@ -7,7 +7,7 @@
 st.markdown(body=gethomepage(), unsafe_allow_html=True)
 
 # Check if user is authenticated
-if not st.experimental_user.is_logged_in:
+if not st.user.is_logged_in:
     if st.button("Log in or Sign up"):
         st.login("auth0")
     st.stop()
@@ -19,22 +19,22 @@
     st.session_state.username = None
 
 # Only access user info if available
-if hasattr(st.experimental_user, "email"):
+if hasattr(st.user, "email"):
     conn = create_connection()
     if conn is not None:
         create_table(conn)
 
         # Check if user exists
-        user_data = get_user(conn, st.experimental_user.email)
+        user_data = get_user(conn, st.user.email)
         if user_data is not None:
             st.session_state['user_type'] = user_data[3]
             st.session_state['username'] = user_data[1]
         else:
             user_type = 'guest'
-            username = st.experimental_user.email
+            username = st.user.email
             st.session_state['user_type'] = user_type
             st.session_state['username'] = username
-            insert_user(conn, username, st.experimental_user.email, user_type)
+            insert_user(conn, username, st.user.email, user_type)
 
 # Show menu
 menu()

pages/chatbot.py

@@ -1,4 +1,4 @@
-from utils import load_yaml_file
+from utils import load_yaml_file, escape_markdown
 from main import get_ragchain
 import streamlit as st
 from menu import menu_with_redirect
@@ -46,7 +46,8 @@
 # -------------------------------
 for message in user_chat:
     with st.chat_message(message["role"], avatar=logo_path if message["role"] == "assistant" else None):
-        st.write(message["content"])
+        #st.write(message["content"])
+        st.markdown(escape_markdown(message["content"]))
 
 # -------------------------------
 # Handle user input
@@ -67,7 +68,8 @@
             response = rag_chain.invoke({"input": query})
             # save response in a text file
             print(response, file=open('responses.txt', 'a', encoding='utf-8'))
-            st.markdown(response["answer"])
+            # Display the response escaping markdown special characters
+            st.markdown(escape_markdown(response["answer"]))
 
         reply_msg = {"role": "assistant", "content": response["answer"]}
         user_chat.append(reply_msg)

utils.py

@@ -269,3 +269,7 @@ def save_transcript(video_id, output_folder):
 
     except Exception as e:
         print(f"Error reading transcript: {e}")
+
+# Function to escape markdown special characters
+def escape_markdown(text: str) -> str:
+    return re.sub(r'([\\`*_{}\[\]()#+\-!$])', r'\\\1', text)