Update app.py

app.py

@@ -178,27 +178,83 @@ def chat_with_model(user_message, chat_history):
 def reset_chat():
     return []
 
+import gradio as gr
+
 def build_ui():
-    custom_css = """
-    #chatbot { background-color:#10121a; color:#e6eef8; border-radius:10px; padding:10px; }
-    """
-    with gr.Blocks(theme=gr.themes.Soft(), ) as demo:
-   
-        chatbot = gr.Chatbot(height=540, elem_id="chatbot", type="tuples")
+    with gr.Blocks(
+        theme=gr.themes.Soft(primary_hue="indigo"),
+        css="""
+            #ohamlab {
+                height: 550px !important;
+                border-radius: 14px;
+                box-shadow: 0 2px 12px rgba(0,0,0,0.1);
+                background: white;
+            }
+            .gradio-container {
+                max-width: 900px !important;
+                margin: auto;
+                padding-top: 1.5rem;
+            }
+            textarea {
+                resize: none !important;
+                border-radius: 12px !important;
+                border: 1px solid #d1d5db !important;
+                box-shadow: 0 1px 3px rgba(0,0,0,0.08);
+            }
+            button.primary {
+                background-color: #4f46e5 !important;
+                color: white !important;
+                border-radius: 10px !important;
+                padding: 0.6rem 1.4rem !important;
+                font-weight: 600;
+                transition: all 0.2s ease-in-out;
+            }
+            button.primary:hover {
+                background-color: #4338ca !important;
+            }
+            button.secondary {
+                background-color: #f3f4f6 !important;
+                border-radius: 10px !important;
+                color: #374151 !important;
+                font-weight: 500;
+                transition: all 0.2s ease-in-out;
+            }
+            button.secondary:hover {
+                background-color: #e5e7eb !important;
+            }
+        """
+    ) as demo:
+
+        gr.Markdown(
+            """
+            <div style="text-align:center; margin-bottom: 1rem;">
+                <h2 style="font-weight:700; font-size:1.8rem;">💠 OhamLab AI Assistant</h2>
+                <p style="color:#6b7280; font-size:0.95rem;">Your intelligent R&D and engineering companion powered by contextual knowledge.</p>
+            </div>
+            """
+        )
+
+        chatbot = gr.Chatbot(height=520, elem_id="ohamlab", type="tuples")
 
         with gr.Row():
             msg = gr.Textbox(
-                placeholder="Type a message…", lines=3, scale=8, show_label=False
+                placeholder="Type your message here...",
+                lines=3,
+                scale=12,
+                show_label=False,
+                container=False,
             )
-            send = gr.Button("Send", variant="primary", scale=1)
-        with gr.Row():
-            clear = gr.Button("Clear")
 
+        with gr.Row(equal_height=True):
+            send = gr.Button("Send", variant="primary", elem_classes=["primary"])
+            clear = gr.Button("Clear", variant="secondary", elem_classes=["secondary"])
+
+        # Wire events
         send.click(chat_with_model, inputs=[msg, chatbot], outputs=[chatbot, msg])
         msg.submit(chat_with_model, inputs=[msg, chatbot], outputs=[chatbot, msg])
         clear.click(reset_chat, outputs=chatbot)
 
-        demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
+    return demo
 
 # ---------------------------
 # Entrypoint