Update app.py

app.py

@@ -58,6 +58,18 @@ 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):
+    parts = []
+    for u, a in (history or []):
+        if u:
+            parts.append(f"User: {u}")
+        if a:
+            parts.append(f"Assistant: {a}")
+    parts.append(f"User: {message}")
+    parts.append("Assistant:")
+    return "\n".join(parts)
+
 # -------------------
 # Cohere Hosted
 # -------------------
@@ -65,31 +77,24 @@ _co_client = None
 if USE_HOSTED_COHERE:
     _co_client = cohere.Client(api_key=COHERE_API_KEY)
 
-def _cohere_parse(resp):
-    if hasattr(resp, "output_text") and resp.output_text:
-        return resp.output_text.strip()
-    if getattr(resp, "message", None) and getattr(resp.message, "content", None):
-        for p in resp.message.content:
-            if hasattr(p, "text") and p.text:
-                return p.text.strip()
-    if hasattr(resp, "text") and resp.text:
-        return resp.text.strip()
-    return "Sorry, I couldn't parse the response from Cohere."
-
 def cohere_chat(message, history):
     try:
-        msgs = []
-        for u, a in (history or []):
-            msgs.append({"role": "user", "content": u})
-            msgs.append({"role": "assistant", "content": a})
-        msgs.append({"role": "user", "content": message})
-        resp = _co_client.responses.create(
+        # 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",
-            messages=msgs,
+            message=prompt,
             temperature=0.3,
             max_tokens=350,
         )
-        return _cohere_parse(resp)
+        # 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:
+            return resp.reply.strip()
+        if hasattr(resp, "generations") and resp.generations:
+            return resp.generations[0].text.strip()
+        return "Sorry, I couldn't parse the response from Cohere."
     except Exception as e:
         return f"Error calling Cohere API: {e}"
 
@@ -99,9 +104,7 @@ def cohere_chat(message, history):
 @lru_cache(maxsize=1)
 def load_local_model():
     if not HF_TOKEN:
-        raise RuntimeError(
-            "HUGGINGFACE_HUB_TOKEN is not set."
-        )
+        raise RuntimeError("HUGGINGFACE_HUB_TOKEN is not set.")
     login(token=HF_TOKEN, add_to_git_credential=False)
     dtype, device_map = pick_dtype_and_map()
     tok = AutoTokenizer.from_pretrained(
@@ -166,7 +169,7 @@ def chat_fn(message, history, user_tz):
         return f"Error: {e}"
 
 # -------------------
-# Theme & CSS
+# Theme & CSS (both bot and user bubbles teal + white text)
 # -------------------
 theme = gr.themes.Soft(
     primary_hue="teal",
@@ -196,6 +199,11 @@ h1 {
   background: var(--brand-accent) !important;
   color: var(--brand-text-light) !important;
 }
+
+/* Inputs a bit softer */
+textarea, input, .gr-input {
+  border-radius: 12px !important;
+}
 """
 
 # -------------------