Update app.py

app.py

@@ -153,12 +153,26 @@ def main():
         # Show history
         for message in st.session_state.messages:
             with st.chat_message(message["role"]):
-                st.write(message["content"])
+                st.markdown(
+                    f"""
+                    <div style="direction: rtl; text-align: right; font-size: 16px;">
+                        {message["content"]}
+                    </div>
+                    """,
+                    unsafe_allow_html=True
+                )
 
         if prompt := st.chat_input():
             st.session_state.messages.append({"role": "user", "content": prompt})
             with st.chat_message("user"):
-                st.write(prompt)
+                st.markdown(
+                    f"""
+                    <div style="direction: rtl; text-align: right; font-size: 16px;">
+                        {prompt}
+                    </div>
+                    """,
+                    unsafe_allow_html=True
+                )
 
             if st.session_state.messages[-1]["role"] != "assistant":
                 try:
@@ -166,7 +180,14 @@ def main():
                         response = user_input(prompt)
                         if response:
                             full_response = response['output_text']
-                            st.write(full_response)
+                            st.markdown(
+                                f"""
+                                <div style="direction: rtl; text-align: right; font-size: 16px;">
+                                    {full_response}
+                                </div>
+                                """,
+                                unsafe_allow_html=True
+                            )
                             st.session_state.messages.append({"role": "assistant", "content": full_response})
                 except RuntimeError as e:
                     st.error(str(e))