Update app.py

app.py

@@ -53,7 +53,7 @@ if uploaded_file and "document_uploaded" not in st.session_state:
     st.session_state["document_uploaded"] = True
 
 # Chat UI Header
-st.title("💬 SOP QA")
+st.title("💬 AI Chat Assistant")
 st.divider()
 
 # Display chat history
@@ -63,7 +63,8 @@ for message in memory_storage.messages:
 
     with st.chat_message(role, avatar=f"data:image/jpeg;base64,{avatar}"):
         if role == "assistant":
-            st.code(message.content, language="markdown")  # Copiable markdown
+            # Display AI response with Markdown formatting for a ChatGPT-like feel
+            st.markdown(f"```\n{message.content}\n```")  
         else:
             st.markdown(message.content)
 
@@ -86,6 +87,6 @@ if user_input:
 
     memory_storage.add_ai_message(answer)
 
-    # Display AI response as a copiable markdown block
+    # Display AI response in a ChatGPT-like format
     with st.chat_message("assistant", avatar=f"data:image/jpeg;base64,{ai_avatar}"):
-        st.code(answer.content, language="markdown")
+        st.markdown(f"```\n{answer.content}\n```")