Update app.py

app.py

@@ -152,6 +152,41 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=512):
 # ✅ Streamlit UI
 st.title("🚀 HAL - NASA AI Assistant")
 
+# ✅ Justify all chatbot responses
+st.markdown("""
+    <style>
+    .user-msg {
+        background-color: #696969;
+        color: white;
+        padding: 10px;
+        border-radius: 10px;
+        margin-bottom: 5px;
+        width: fit-content;
+        max-width: 80%;
+        text-align: justify;  /* ✅ Justify text */
+    }
+    .assistant-msg {
+        background-color: #333333;
+        color: white;
+        padding: 10px;
+        border-radius: 10px;
+        margin-bottom: 5px;
+        width: fit-content;
+        max-width: 80%;
+        text-align: justify;  /* ✅ Justify text */
+    }
+    .container {
+        display: flex;
+        flex-direction: column;
+        align-items: flex-start;
+    }
+    @media (max-width: 600px) {
+        .user-msg, .assistant-msg { font-size: 16px; max-width: 100%; }
+    }
+    </style>
+""", unsafe_allow_html=True)
+
+
 # ✅ Reset Chat Button
 if st.sidebar.button("Reset Chat"):
     st.session_state.chat_history = [{"role": "assistant", "content": "Hello! How can I assist you today?"}]