Update app.py

app.py

@@ -106,9 +106,41 @@ st.markdown("🌌 *Powered by Hugging Face & NASA APIs!*")
 # Custom CSS for chat styling
 st.markdown("""
     <style>
-    .user-msg { background-color: #ADD8E6; padding: 10px; border-radius: 10px; margin-bottom: 5px; }
-    .assistant-msg { background-color: #F0F0F0; padding: 10px; border-radius: 10px; margin-bottom: 5px; }
-    @media (max-width: 600px) { .user-msg, .assistant-msg { font-size: 16px; } }
+    /* Style for chat messages */
+    .user-msg {
+        background-color: #0078D7; /* Dark Blue */
+        color: white; /* White text for contrast */
+        padding: 10px;
+        border-radius: 10px;
+        margin-bottom: 5px;
+        width: fit-content;
+        max-width: 80%;
+    }
+
+    .assistant-msg {
+        background-color: #333333; /* Dark Gray */
+        color: white; /* White text for contrast */
+        padding: 10px;
+        border-radius: 10px;
+        margin-bottom: 5px;
+        width: fit-content;
+        max-width: 80%;
+    }
+
+    /* Center messages for better appearance */
+    .container {
+        display: flex;
+        flex-direction: column;
+        align-items: flex-start;
+    }
+
+    /* Adjust messages on mobile */
+    @media (max-width: 600px) {
+        .user-msg, .assistant-msg { 
+            font-size: 16px;
+            max-width: 100%;
+        }
+    }
     </style>
 """, unsafe_allow_html=True)