Update app.py

app.py

@@ -65,11 +65,16 @@ def pick_dtype_and_map():
 
 def is_identity_query(message, history):
     patterns = [
-        r"\bwho\s+are\s+you\b", r"\bwhat\s+are\s+you\b",
-        r"\bwhat\s+is\s+your\s+name\b", r"\bwho\s+is\s+this\b",
-        r"\bidentify\s+yourself\b", r"\btell\s+me\s+about\s+yourself\b",
-        r"\bdescribe\s+yourself\b", r"\band\s+you\s*\?\b",
-        r"\byour\s+name\b", r"\bwho\s+am\s+i\s+chatting\s+with\b"
+        r"\bwho\s+are\s+you\b",
+        r"\bwhat\s+are\s+you\b",
+        r"\bwhat\s+is\s+your\s+name\b",
+        r"\bwho\s+is\s+this\b",
+        r"\bidentify\s+yourself\b",
+        r"\btell\s+me\s+about\s+yourself\b",
+        r"\bdescribe\s+yourself\b",
+        r"\band\s+you\s*\?\b",
+        r"\byour\s+name\b",
+        r"\bwho\s+am\s+i\s+chatting\s+with\b"
     ]
     def match(t):
         return any(re.search(p, (t or "").strip().lower()) for p in patterns)
@@ -82,8 +87,7 @@ def is_identity_query(message, history):
     return False
 
 def _iter_user_assistant(history):
-    """
-    Yield (user, assistant) pairs from a Gradio history list.
+    """Yield (user, assistant) pairs from a Gradio history list.
     Safely handles items that are lists/tuples with >2 elements.
     """
     for item in (history or []):
@@ -91,7 +95,7 @@ def _iter_user_assistant(history):
             u = item[0] if len(item) > 0 else ""
             a = item[1] if len(item) > 1 else ""
             yield u, a
-        # If dicts ever appear, extend handling here.
+    # If dicts ever appear, extend handling here.
 
 def _history_to_prompt(message, history):
     """Build a simple text prompt for the stable cohere.chat API."""
@@ -189,11 +193,11 @@ def local_generate(model, tokenizer, input_ids, max_new_tokens=350):
     return tokenizer.decode(gen_only, skip_special_tokens=True).strip()
 
 # -------------------
-# Chat Function (now with Safety layer)
+# Chat Function (with Safety layer)
 # -------------------
 def chat_fn(message, history, user_tz):
     try:
-        # ---- INPUT SAFETY (your snippet semantics) ----
+        # ---- INPUT SAFETY ----
         safe_in, blocked_in, reason_in = safety_filter(message, mode="input")
         if blocked_in:
             return refusal_reply(reason_in)
@@ -210,7 +214,7 @@ def chat_fn(message, history, user_tz):
             inputs = build_inputs(tokenizer, safe_in, history)
             out = local_generate(model, tokenizer, inputs, max_new_tokens=350)
 
-        # Tidy echoes (from your snippet)
+        # Tidy echoes
         if isinstance(out, str):
             for tag in ("Assistant:", "System:", "User:"):
                 if out.startswith(tag):
@@ -220,7 +224,6 @@ def chat_fn(message, history, user_tz):
         safe_out, blocked_out, reason_out = safety_filter(out, mode="output")
         if blocked_out:
             return refusal_reply(reason_out)
-
         return safe_out
     except Exception as e:
         return f"Error: {e}"
@@ -241,35 +244,38 @@ custom_css = """
   --brand-text: #0f172a;
   --brand-text-light: #ffffff;
 }
+
 /* Page background */
-.gradio-container {
-  background: var(--brand-bg);
-}
+.gradio-container { background: var(--brand-bg); }
+
 /* Title */
 h1 {
   color: var(--brand-text);
   font-weight: 700;
   font-size: 28px !important;
 }
-/* Try to hide the default Chatbot label via CSS for multiple Gradio builds */
-.chatbot header,
-.chatbot .label,
-.chatbot .label-wrap,
-.chatbot .top,
-.chatbot .header,
-.chatbot > .wrap > header {
+
+/* Hide default Chatbot label (cover most Gradio builds) */
+.chatbot header, .chatbot .label, .chatbot .label-wrap, .chatbot .top, .chatbot .header, .chatbot > .wrap > header {
   display: none !important;
 }
-/* Both bot and user bubbles teal with white text */
+
+/* Bubble styling */
 .message.user, .message.bot {
   background: var(--brand-accent) !important;
   color: var(--brand-text-light) !important;
   border-radius: 12px !important;
   padding: 8px 12px !important;
 }
+
 /* Inputs a bit softer */
-textarea, input, .gr-input {
-  border-radius: 12px !important;
+textarea, input, .gr-input { border-radius: 12px !important; }
+
+/* Center examples */
+.examples, .examples .grid {
+  display: flex !important;
+  justify-content: center !important;
+  text-align: center !important;
 }
 """
 
@@ -279,8 +285,12 @@ textarea, input, .gr-input {
 with gr.Blocks(theme=theme, css=custom_css) as demo:
     # Hidden box to carry timezone (still useful for future features)
     tz_box = gr.Textbox(visible=False)
-    demo.load(lambda tz: tz, inputs=[tz_box], outputs=[tz_box],
-              js="() => Intl.DateTimeFormat().resolvedOptions().timeZone")
+    demo.load(
+        lambda tz: tz,
+        inputs=[tz_box],
+        outputs=[tz_box],
+        js="() => Intl.DateTimeFormat().resolvedOptions().timeZone",
+    )
 
     # Extra JS hard-removal of the Chatbot label to cover all DOM variants
     hide_label_sink = gr.HTML(visible=False)
@@ -289,35 +299,45 @@ with gr.Blocks(theme=theme, css=custom_css) as demo:
         inputs=None,
         outputs=hide_label_sink,
         js="""
-() => {
-  const sel = [
-    '.chatbot header',
-    '.chatbot .label',
-    '.chatbot .label-wrap',
-    '.chatbot .top',
-    '.chatbot .header',
-    '.chatbot > .wrap > header'
-  ];
-  sel.forEach(s => document.querySelectorAll(s).forEach(el => el.style.display = 'none'));
-  return "";
-}
-"""
+        () => {
+          const sel = [
+            '.chatbot header',
+            '.chatbot .label',
+            '.chatbot .label-wrap',
+            '.chatbot .top',
+            '.chatbot .header',
+            '.chatbot > .wrap > header'
+          ];
+          sel.forEach(s => document.querySelectorAll(s).forEach(el => el.style.display = 'none'));
+          return "";
+        }
+        """,
     )
 
-    # Updated title
+    # Title
     gr.Markdown("# ClarityOps Augmented Decision AI")
 
+    # Chat interface (larger chat, no Undo; examples centered & single-column)
     gr.ChatInterface(
         fn=chat_fn,
         type="messages",
         additional_inputs=[tz_box],
-        chatbot=gr.Chatbot(label="", show_label=False, type="messages"),  # aligned type + no label
+        chatbot=gr.Chatbot(
+            label="",
+            show_label=False,
+            type="messages",
+            height=700,  # larger chat window
+        ),
         examples=[
-            ["What are the symptoms of hypertension?", ""],
-            ["What are common drug interactions with aspirin?", ""],
-            ["What are the warning signs of diabetes?", ""],
+            ["What are the symptoms of hypertension?"],
+            ["What are common drug interactions with aspirin?"],
+            ["What are the warning signs of diabetes?"],
         ],
-        cache_examples=False,  # prevent permission error in Spaces
+        cache_examples=False,
+        submit_btn="Submit",
+        retry_btn="Retry",
+        clear_btn="Clear",
+        undo_btn=None,  # removed Undo button
     )
 
 if __name__ == "__main__":
@@ -326,7 +346,8 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=port,
-        show_api=False,      # optional: less overhead
-        max_threads=8,       # optional: avoid thread-starvation on tiny CPUs
+        show_api=False,  # optional: less overhead
+        max_threads=8,   # optional: avoid thread-starvation on tiny CPUs
     )
 
+