update

src/pages/{5_Ask_Streaming.py → 1_Ask_Streaming.py}

@@ -75,9 +75,9 @@ def main():
     for message in st.session_state.streaming_messages:
         with st.chat_message(message["role"]):
             if message["role"] == "user":
-                st.write(message["content"])
+                st.text(message["content"])
             else:
-                st.write(message["content"])
+                st.text(message["content"])
                 if "sql" in message:
                     with st.expander("= Generated SQL Query", expanded=False):
                         st.code(format_sql(message["sql"]), language="sql")
@@ -148,7 +148,7 @@ def main():
                         delta = event.get("delta", {})
                         if delta.get("type") == "text_delta":
                             final_answer += delta.get("text", "")
-                            content_container.markdown(final_answer)
+                            content_container.text(final_answer)
                     
                     elif event.get("type") == "message_stop":
                         # Extract final data from the event
@@ -184,7 +184,7 @@ def main():
                         assistant_message["sql"] = final_sql
                     
                     st.session_state.streaming_messages.append(assistant_message)
-                    content_container.write(final_answer)
+                    content_container.text(final_answer)
 
                     # Display SQL query if available
                     if final_sql:
@@ -214,7 +214,7 @@ def main():
                         "content": "I couldn't generate an answer for your question. Please try rephrasing it or make sure it's related to your data."
                     }
                     st.session_state.streaming_messages.append(assistant_message)
-                    content_container.write(assistant_message["content"])
+                    content_container.text(assistant_message["content"])
             
             except Exception as e:
                 st.error(f"Error during streaming: {str(e)}")
@@ -223,7 +223,7 @@ def main():
                     "content": "Sorry, I couldn't process your request. Please check your API credentials and try again."
                 }
                 st.session_state.streaming_messages.append(assistant_message)
-                content_container.write(assistant_message["content"])
+                content_container.text(assistant_message["content"])
 
     # Clear chat button
     if st.sidebar.button("🧹 Clear Chat History"):

src/pages/{1_Query_And_Answer.py → 2_SQL_Generation.py}

@@ -6,7 +6,7 @@ from utils import format_sql
 
 
 def main():
-    st.title("💬 Wren AI Cloud API Demo - Query and Answer")
+    st.title("💬 Wren AI Cloud API Demo - SQL Generation")
 
     if "api_key" not in st.session_state or "project_id" not in st.session_state:
         st.error("Please enter your API Key and Project ID in the sidebar of Home page to get started.")