Update app.py

app.py

@@ -28,7 +28,29 @@ def chat_function(message, history):
 
     return "", history
 
-with gr.Blocks(theme=gr.themes.Glass(), css=".gradio-container .gradio-input label, .gradio-container .gradio-output label { display: none; }") as demo:
+with gr.Blocks(theme=gr.themes.Glass(), css="""
+    .gradio-chatbot .message.user {
+        background-color: #e1f7d5;
+        border-radius: 10px;
+        padding: 10px;
+        margin: 5px;
+        max-width: 70%;
+        margin-left: auto;
+    }
+
+    .gradio-chatbot .message.ai {
+        background-color: #f0f0f0;
+        border-radius: 10px;
+        padding: 10px;
+        margin: 5px;
+        max-width: 70%;
+        margin-right: auto;
+    }
+
+    .gradio-chatbot .message {
+        font-size: 16px;
+    }
+""") as demo:
     chatbot = gr.Chatbot(label=None)
     msg = gr.Textbox(placeholder="Message Z3ta...", label=None)