Update styles.py

styles.py

@@ -81,5 +81,14 @@ def get_custom_css():
             border-color: #3D89B8;
             border-radius: 5px;
         }
+        
+        /* Add border to the model response area */
+        [data-testid="column"][data-column-index="1"] > div {
+            border: 1px solid #3D89B8;
+            border-radius: 8px;
+            padding: 15px;
+            margin-top: 10px;
+            box-shadow: 0 2px 5px rgba(61, 137, 184, 0.1);
+        }
     </style>
     """