Update app.py

app.py

@@ -188,26 +188,26 @@ def format_chat_history(history):
 def display_chat_messages():
     """Display chat messages in a conversational format."""
     for message in st.session_state.chat_history:
-        # User message - right aligned
+        # User message - right aligned with wider column
         with st.container():
-            col1, col2 = st.columns([6, 4])
+            col1, col2 = st.columns([2, 8])  # Changed from [6, 4] to [2, 8]
             with col2:
                 st.markdown(f"""
                     <div style='background-color: #007AFF; color: white; padding: 12px 16px; 
-                        border-radius: 15px 15px 0 15px; margin-bottom: 8px; text-align: right;
-                        box-shadow: 0 1px 2px rgba(0,0,0,0.1);'>
+                        border-radius: 15px 15px 0 15px; margin-bottom: 8px; text-align: left;
+                        box-shadow: 0 1px 2px rgba(0,0,0,0.1); max-width: 800px;'>
                         {message['question']}
                     </div>
                     """, unsafe_allow_html=True)
         
-        # Assistant message - left aligned
+        # Assistant message - left aligned with wider column
         with st.container():
-            col1, col2 = st.columns([4, 6])
+            col1, col2 = st.columns([8, 2])  # Changed from [4, 6] to [8, 2]
             with col1:
                 st.markdown(f"""
                     <div style='background-color: #f0f2f6; padding: 12px 16px; 
                         border-radius: 15px 15px 15px 0; margin-bottom: 16px;
-                        box-shadow: 0 1px 2px rgba(0,0,0,0.1);'>
+                        box-shadow: 0 1px 2px rgba(0,0,0,0.1); max-width: 800px;'>
                         {message['answer']}
                     </div>
                     """, unsafe_allow_html=True)