Update app.py

app.py

@@ -153,12 +153,6 @@ CSS = """
   background: var(--bg-secondary);
 }
 
-.gradio-chatbot {
-  min-height: 100%;
-  background: transparent !important;
-  border: none !important;
-}
-
 /* Input Area */
 #input-area {
   padding: 24px;
@@ -243,27 +237,27 @@ CSS = """
 }
 
 /* Chat Messages Customization */
-.user-message, .assistant-message {
-  padding: 16px 20px;
-  border-radius: 18px;
-  margin: 8px 0;
-  max-width: 85%;
-  line-height: 1.5;
+.user[data-testid="user"], .assistant[data-testid="assistant"] {
+  padding: 16px 20px !important;
+  border-radius: 18px !important;
+  margin: 8px 0 !important;
+  max-width: 85% !important;
+  line-height: 1.5 !important;
 }
 
-.user-message {
-  background: var(--primary-color);
-  color: white;
-  margin-left: auto;
-  border-bottom-right-radius: 4px;
+.user[data-testid="user"] {
+  background: var(--primary-color) !important;
+  color: white !important;
+  margin-left: auto !important;
+  border-bottom-right-radius: 4px !important;
 }
 
-.assistant-message {
-  background: var(--bg-secondary);
-  color: var(--text-primary);
-  margin-right: auto;
-  border-bottom-left-radius: 4px;
-  border: 1px solid var(--border-color);
+.assistant[data-testid="assistant"] {
+  background: var(--bg-secondary) !important;
+  color: var(--text-primary) !important;
+  margin-right: auto !important;
+  border-bottom-left-radius: 4px !important;
+  border: 1px solid var(--border-color) !important;
 }
 
 /* Model Info Badge */
@@ -299,12 +293,13 @@ CSS = """
     flex-wrap: wrap;
   }
 }
-"""
 
-theme = gr.themes.Soft(
-    primary_hue="indigo",
-    secondary_hue="slate"
-)
+/* Hide default Gradio borders */
+.gr-chatbot {
+  border: none !important;
+  background: transparent !important;
+}
+"""
 
 # ========= HELPERS =========
 def normalize_whitespace(s: str) -> str:
@@ -423,7 +418,7 @@ def on_clear():
     return [], "", "No files loaded"
 
 # ========= MODERN UI =========
-with gr.Blocks(title="iChat AI Assistant", css=CSS, theme=theme) as demo:
+with gr.Blocks(title="iChat AI Assistant") as demo:
     file_context_state = gr.State("")
     chat_state = gr.State([])
     
@@ -499,10 +494,8 @@ with gr.Blocks(title="iChat AI Assistant", css=CSS, theme=theme) as demo:
             chatbot = gr.Chatbot(
                 value=[],
                 elem_id="chatbot-container",
-                height=600,
-                bubble_full_width=False,
-                show_copy_button=True,
-                avatar_images=(None, "https://api.dicebear.com/7.x/bottts-neutral/svg?seed=AI"))
+                height=600
+            )
             
             # Input Area
             with gr.Column(elem_id="input-area"):
@@ -595,4 +588,4 @@ with gr.Blocks(title="iChat AI Assistant", css=CSS, theme=theme) as demo:
         outputs=[file_context_state, file_info_display]
     )
 
-demo.launch()
+demo.launch(css=CSS)