Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import re
-from datetime import datetime, timezone
 from functools import lru_cache
+from datetime import datetime, timezone
 
 import gradio as gr
 import torch
@@ -12,7 +12,7 @@ try:
 except Exception:
     ZoneInfo = None
 
-# Cohere SDK
+# Cohere SDK (hosted path)
 try:
     import cohere
     _HAS_COHERE = True
@@ -58,8 +58,8 @@ def is_identity_query(message, history):
             return True
     return False
 
-# Build a simple prompt string for the older/stable Cohere chat API.
 def _history_to_prompt(message, history):
+    """Build a simple text prompt for the stable cohere.chat API."""
     parts = []
     for u, a in (history or []):
         if u:
@@ -79,7 +79,6 @@ if USE_HOSTED_COHERE:
 
 def cohere_chat(message, history):
     try:
-        # Use the stable chat() API to avoid SDK breaking changes.
         prompt = _history_to_prompt(message, history)
         resp = _co_client.chat(
             model="command-r7b-12-2024",
@@ -87,7 +86,6 @@ def cohere_chat(message, history):
             temperature=0.3,
             max_tokens=350,
         )
-        # Newer SDKs provide .text; older ones may use .reply or generations
         if hasattr(resp, "text") and resp.text:
             return resp.text.strip()
         if hasattr(resp, "reply") and resp.reply:
@@ -169,7 +167,7 @@ def chat_fn(message, history, user_tz):
         return f"Error: {e}"
 
 # -------------------
-# Theme & CSS (both bot and user bubbles teal + white text)
+# Theme & CSS
 # -------------------
 theme = gr.themes.Soft(
     primary_hue="teal",
@@ -181,19 +179,31 @@ custom_css = """
 :root {
   --brand-bg: #e6f7f8; /* soft medical teal */
   --brand-accent: #0d9488; /* teal-600 */
+  --brand-text: #0f172a;
   --brand-text-light: #ffffff;
 }
 
+/* Page background */
 .gradio-container {
   background: var(--brand-bg);
 }
 
+/* Title */
 h1 {
-  color: #0f172a;
+  color: var(--brand-text);
   font-weight: 700;
   font-size: 28px !important;
 }
 
+/* Hide the default "Chatbot" label pill in the top-left of the chat */
+.chatbot .label,
+.chatbot .label-wrap,
+.chatbot header,
+.chatbot .top,
+.chatbot .header {
+  display: none !important;
+}
+
 /* Both bot and user bubbles teal with white text */
 .message.user, .message.bot {
   background: var(--brand-accent) !important;
@@ -210,11 +220,12 @@ textarea, input, .gr-input {
 # UI
 # -------------------
 with gr.Blocks(theme=theme, css=custom_css) as demo:
+    # Capture browser timezone (stored but not yet used)
     tz_box = gr.Textbox(visible=False)
     demo.load(lambda tz: tz, inputs=[tz_box], outputs=[tz_box],
               js="() => Intl.DateTimeFormat().resolvedOptions().timeZone")
 
-    gr.Markdown("# ClarityOps Augmented Decision AI")
+    gr.Markdown("# Medical Decision Support AI")
 
     gr.ChatInterface(
         fn=chat_fn,
@@ -234,3 +245,4 @@ if __name__ == "__main__":
 
 
 
+