Update app.py

app.py

@@ -128,7 +128,6 @@ def cohere_chat(message, history):
         return None
     try:
         client = cohere.Client(api_key=COHERE_API_KEY)
-        # Build a simple conversational prompt (history included)
         parts = []
         for u, a in _iter_user_assistant(history):
             if u: parts.append(f"User: {u}")
@@ -269,11 +268,6 @@ def _is_mdsi_diabetes(text: str) -> bool:
     return any(k in t for k in ["mdsi", "mobile diabetes", "diabetes", "metabolic", "a1c", "metis"])
 
 def build_dynamic_clarifications(scenario_text: str, artifacts: List[Dict[str, Any]]) -> str:
-    """
-    Build up to 5 grouped clarification questions based on what's MISSING.
-    Groups: Prioritization, Capacity, Cost, Clinical, Recommendations.
-    Only ask for domains not covered by uploads/scenario text.
-    """
     flags_from_files = _extract_present_domains(artifacts)
     flags_from_text = _domain_from_text(scenario_text)
     missing = {
@@ -335,26 +329,18 @@ def build_dynamic_clarifications(scenario_text: str, artifacts: List[Dict[str, A
 
 # ---------- Core chat logic (auto scenario, dynamic Phase 1) ----------
 def clarityops_reply(user_msg, history, tz, uploaded_files_paths, awaiting_answers=False):
-    """
-    awaiting_answers:
-      - False: If scenario triggered -> Phase 1 (dynamic questions). Else normal chat.
-      - True:  If scenario triggered -> Phase 2 (structured analysis). Else normal chat.
-    """
     try:
         log_event("user_message", None, {"sizes": {"chars": len(user_msg or "")}})
 
-        # Safety (input)
         safe_in, blocked_in, reason_in = safety_filter(user_msg, mode="input")
         if blocked_in:
             ans = refusal_reply(reason_in)
             return history + [(user_msg, ans)], awaiting_answers
 
-        # Identity short-circuit
         if is_identity_query(safe_in, history):
             ans = "I am ClarityOps, your strategic decision making AI partner."
             return history + [(user_msg, ans)], awaiting_answers
 
-        # Ingest uploads FIRST (files alone can trigger scenario mode)
         artifacts = []
         if uploaded_files_paths:
             ing = extract_text_from_files(uploaded_files_paths)
@@ -366,17 +352,14 @@ def clarityops_reply(user_msg, history, tz, uploaded_files_paths, awaiting_answe
                 _session_rag.register_artifacts(artifacts)
             log_event("uploads_added", None, {"chunks": len(chunks), "artifacts": len(artifacts)})
 
-        # CSV columns helper (works in both modes)
         if re.search(r"\b(columns?|headers?)\b", (safe_in or "").lower()):
             cols = _session_rag.get_latest_csv_columns()
             if cols:
                 return history + [(user_msg, "Here are the column names from your most recent CSV upload:\n\n- " + "\n- ".join(cols))], awaiting_answers
 
-        # Decide mode
         scenario_mode = is_scenario_triggered(safe_in, uploaded_files_paths)
 
         if not scenario_mode:
-            # ---------- Normal conversational chat ----------
             out = cohere_chat(safe_in, history) if USE_HOSTED_COHERE else None
             if not out:
                 model, tokenizer = load_local_model()
@@ -399,9 +382,7 @@ def clarityops_reply(user_msg, history, tz, uploaded_files_paths, awaiting_answe
             })
             return history + [(user_msg, safe_out)], awaiting_answers
 
-        # ---------- Scenario Mode ----------
         if not awaiting_answers:
-            # PHASE 1: generate dynamic questions here (no assumptions)
             phase1 = build_dynamic_clarifications(scenario_text=safe_in, artifacts=artifacts or _session_rag.artifacts)
             phase1 = _sanitize_text(phase1)
             log_event("assistant_reply", None, {
@@ -412,7 +393,6 @@ def clarityops_reply(user_msg, history, tz, uploaded_files_paths, awaiting_answe
             })
             return history + [(user_msg, phase1)], True
 
-        # PHASE 2: build rich system preamble + feed to LLM
         session_snips = "\n---\n".join(_session_rag.retrieve(
             "diabetes screening Indigenous Métis mobile program cost throughput outcomes logistics",
             k=6
@@ -427,7 +407,6 @@ def clarityops_reply(user_msg, history, tz, uploaded_files_paths, awaiting_answe
         user_lower = (safe_in or "").lower()
         mdsi_extra = _mdsi_block() if ("diabetes" in user_lower or "mdsi" in user_lower or "mobile screening" in user_lower) else ""
 
-        # Summarize artifacts for the model (concise, structured)
         arts = _session_rag.artifacts or []
         if arts:
             arts_summ = []
@@ -494,7 +473,7 @@ def clarityops_reply(user_msg, history, tz, uploaded_files_paths, awaiting_answe
 # ---------- Theme & CSS ----------
 theme = gr.themes.Soft(primary_hue="teal", neutral_hue="slate", radius_size=gr.themes.sizes.radius_lg)
 custom_css = """
-:root { --brand-bg: #e6f7f8; --brand-accent: #0d9488; --brand-text: #0f172a; --brand-text-light: #ffffff; }
+:root { --brand-bg: #0f172a; --brand-accent: #0d9488; --brand-text: #0f172a; --brand-text-light: #ffffff; }  /* CHANGED bg only */
 html, body, .gradio-container { height: 100vh; }
 .gradio-container { background: var(--brand-bg); display: flex; flex-direction: column; }
 
@@ -504,7 +483,6 @@ html, body, .gradio-container { height: 100vh; }
 #hero h2 { color: #0f172a; font-weight: 800; font-size: 32px; margin-bottom: 22px; }
 #hero .search-row { width: min(860px, 92vw); margin: 0 auto; display: flex; gap: 8px; align-items: stretch; }
 #hero .search-row .hero-box { flex: 1 1 auto; }
-/* Force equal heights between the single-line textbox and the submit button */
 #hero .search-row .hero-box textarea { height: 52px !important; }
 #hero-send > button { height: 52px !important; padding: 0 18px !important; border-radius: 12px !important; }
 #hero .hint { color: #334155; margin-top: 10px; font-size: 13px; opacity: 0.9; }
@@ -560,7 +538,7 @@ with gr.Blocks(theme=theme, css=custom_css, analytics_enabled=False) as demo:
     # ---- State
     state_history = gr.State(value=[])
     state_uploaded = gr.State(value=[])
-    state_awaiting = gr.State(value=False)  # False -> Phase 1 next; True -> Phase 2 next (awaiting answers)
+    state_awaiting = gr.State(value=False)
 
     # ---- Uploads
     def _store_uploads(files, current):
@@ -592,9 +570,9 @@ with gr.Blocks(theme=theme, css=custom_css, analytics_enabled=False) as demo:
         chat_o, msg_o, hist_o, await_o = _on_send(user_msg, history, up_paths, awaiting)
         return (
             chat_o, msg_o, hist_o, await_o,
-            gr.update(visible=False),   # hide hero
-            gr.update(visible=True),    # show app
-            ""                          # clear hero box
+            gr.update(visible=False),
+            gr.update(visible=True),
+            ""
         )
 
     hero_send.click(
@@ -619,12 +597,11 @@ with gr.Blocks(theme=theme, css=custom_css, analytics_enabled=False) as demo:
                concurrency_limit=2, queue=True)
 
     def _on_clear():
-        # Reset to fresh hero screen
         return (
             [], "", [], False,
-            gr.update(visible=True),   # show hero
-            gr.update(visible=False),  # hide app
-            ""                         # clear hero input
+            gr.update(visible=True),
+            gr.update(visible=False),
+            ""
         )
 
     clear.click(_on_clear, None, [chat, msg, state_history, state_awaiting, hero_wrap, app_wrap, hero_msg])
@@ -633,3 +610,4 @@ if __name__ == "__main__":
     port = int(os.environ.get("PORT", "7860"))
     demo.launch(server_name="0.0.0.0", server_port=port, show_api=False, max_threads=8)
 
+