Update app.py

app.py

@@ -10,17 +10,17 @@ from audit_log import log_event, hash_summary
 from privacy import redact_text
 
 # ---------- Environment / cache (Spaces-safe, writable) ----------
-HOME = pathlib.Path.home()  # /home/user on Spaces
+HOME = pathlib.Path.home()
 HF_HOME = str(HOME / ".cache" / "huggingface")
 HF_HUB_CACHE = str(HOME / ".cache" / "huggingface" / "hub")
 HF_TRANSFORMERS = str(HOME / ".cache" / "huggingface" / "transformers")
 ST_HOME = str(HOME / ".cache" / "sentence-transformers")
-GRADIO_TMP = str(HOME / "app" / "gradio")  # or "/tmp/gradio"
+GRADIO_TMP = str(HOME / "app" / "gradio")
 GRADIO_CACHE = GRADIO_TMP
 
 os.environ.setdefault("HF_HOME", HF_HOME)
 os.environ.setdefault("HF_HUB_CACHE", HF_HUB_CACHE)
-os.environ.setdefault("TRANSFORMERS_CACHE", HF_TRANSFORMERS)  # warning is fine
+os.environ.setdefault("TRANSFORMERS_CACHE", HF_TRANSFORMERS)
 os.environ.setdefault("SENTENCE_TRANSFORMERS_HOME", ST_HOME)
 os.environ.setdefault("GRADIO_TEMP_DIR", GRADIO_TMP)
 os.environ.setdefault("GRADIO_CACHE_DIR", GRADIO_CACHE)
@@ -52,13 +52,13 @@ from session_rag import SessionRAG
 from mdsi_analysis import capacity_projection, cost_estimate, outcomes_summary
 
 # ---------- Config ----------
-MODEL_ID = os.getenv("MODEL_ID", "microsoft/Phi-3-mini-4k-instruct")  # local fallback
+MODEL_ID = os.getenv("MODEL_ID", "microsoft/Phi-3-mini-4k-instruct")  # fallback
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN") or os.getenv("HF_TOKEN")
 
 COHERE_API_KEY = os.getenv("COHERE_API_KEY")
 USE_HOSTED_COHERE = bool(COHERE_API_KEY and _HAS_COHERE)
 
-# BIGGER OUTPUT LIMIT
+# Larger output (Cohere + HF fallback)
 MAX_NEW_TOKENS = int(os.getenv("MAX_NEW_TOKENS", "2048"))
 
 # ---------- System Master (two-phase, LLM-only behavior) ----------
@@ -152,7 +152,7 @@ def cohere_chat(message, history):
             model="command-r7b-12-2024",
             message=prompt,
             temperature=0.3,
-            max_tokens=MAX_NEW_TOKENS,  # uses 2048
+            max_tokens=MAX_NEW_TOKENS,
         )
         if hasattr(resp, "text") and resp.text: return resp.text.strip()
         if hasattr(resp, "reply") and resp.reply: return resp.reply.strip()
@@ -204,7 +204,7 @@ def local_generate(model, tokenizer, input_ids, max_new_tokens=MAX_NEW_TOKENS):
     input_ids = input_ids.to(model.device)
     with torch.no_grad():
         out = model.generate(
-            input_ids=input_ids, max_new_tokens=max_new_tokens,  # uses 2048
+            input_ids=input_ids, max_new_tokens=max_new_tokens,
             do_sample=True, temperature=0.3, top_p=0.9,
             repetition_penalty=1.15,
             pad_token_id=tokenizer.eos_token_id,
@@ -264,7 +264,7 @@ def clarityops_reply(user_msg, history, tz, uploaded_files_paths, awaiting_answe
             ans = "I am ClarityOps, your strategic decision making AI partner."
             return history + [(user_msg, ans)], awaiting_answers
 
-        # Ingest uploads (text + artifacts like CSV headers)
+        # Ingest uploads
         if uploaded_files_paths:
             ing = extract_text_from_files(uploaded_files_paths)
             chunks = ing.get("chunks", []) if isinstance(ing, dict) else (ing or [])
@@ -370,131 +370,130 @@ theme = gr.themes.Soft(primary_hue="teal", neutral_hue="slate", radius_size=gr.t
 custom_css = """
 :root { --brand-bg: #e6f7f8; --brand-accent: #0d9488; --brand-text: #0f172a; --brand-text-light: #ffffff; }
 
-/* Full-height layout */
 html, body, .gradio-container { height: 100vh; }
 .gradio-container { background: var(--brand-bg); display: flex; flex-direction: column; }
 
-/* Tidy typography */
-h1 { color: var(--brand-text); font-weight: 700; font-size: 28px !important; }
+/* HERO (landing) */
+#hero-wrap { height: 70vh; display: grid; place-items: center; }
+#hero { text-align: center; }
+#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; }
+#hero .search-row .hero-box { flex: 1 1 auto; }
+#hero .hint { color: #334155; margin-top: 10px; font-size: 13px; opacity: 0.9; }
 
-/* Minimal chrome */
-.chatbot header, .chatbot .label, .chatbot .label-wrap, .chatbot .top, .chatbot .header, .chatbot > .wrap > header { display: none !important; }
+/* CHAT */
+#chat-container { position: relative; }
 .message.user, .message.bot { background: var(--brand-accent) !important; color: var(--brand-text-light) !important; border-radius: 12px !important; padding: 8px 12px !important; }
+.chatbot header, .chatbot .label, .chatbot .label-wrap { display: none !important; }
 textarea, input, .gr-input { border-radius: 12px !important; }
-
-/* Anchor for overlay */
-#chat-container { position: relative; }
-
-/* Centered handshake overlay INSIDE the chat area */
-#handshake-overlay {
-  position: absolute;
-  top: 0; left: 0; right: 0; bottom: 0;   /* fully cover the chat surface */
-  margin: 8px;                            /* reveal rounded corners of chat */
-  z-index: 9999;                          /* sit above the chat content */
-  display: grid;
-  place-items: center;
-  pointer-events: none;                   /* do not block input below */
-}
-
-#handshake-overlay .panel {
-  pointer-events: auto;                   /* if you later want to click to dismiss */
-  background: rgba(13, 148, 136, 0.96);
-  color: #fff;
-  padding: 18px 22px;
-  border-radius: 14px;
-  font-size: 16px;
-  max-width: 720px;
-  text-align: center;
-  box-shadow: 0 10px 24px rgba(0,0,0,0.2);
-}
-
-#handshake-overlay.hidden { display: none; }
-
-@media (max-height: 700px) {
-  #handshake-overlay .panel { font-size: 14px; padding: 14px 16px; max-width: 90vw; }
-}
 """
 
 # ---------- UI ----------
 with gr.Blocks(theme=theme, css=custom_css, analytics_enabled=False) as demo:
-    gr.Markdown("# ClarityOps Augmented Decision AI")
-
-    with gr.Column(elem_id="chat-container"):
-        chat = gr.Chatbot(label="", show_label=False, height="62vh")
-        handshake = gr.HTML(
-            value=(
-                '<div id="handshake-overlay">'
-                '<div class="panel">ClarityOps loaded. Paste your scenario and attach files. '
-                'I’ll ask up to 5 clarifications, then produce the structured analysis</div>'
-                '</div>'
+    # --- HERO (initial Google-like screen) ---
+    with gr.Column(elem_id="hero-wrap", visible=True) as hero_wrap:
+        with gr.Column(elem_id="hero"):
+            gr.HTML("<h2>What can I help with?</h2>")
+            with gr.Row(elem_classes="search-row"):
+                hero_msg = gr.Textbox(
+                    placeholder="Ask anything (paste scenarios here; you can attach files after)...",
+                    show_label=False,
+                    lines=1,
+                    elem_classes="hero-box"
+                )
+                hero_send = gr.Button("➤", scale=0)
+            gr.Markdown('<div class="hint">ClarityOps will first ask up to 5 clarifications, then produce a structured analysis.</div>')
+
+    # --- MAIN APP (hidden until first message) ---
+    with gr.Column(elem_id="chat-container", visible=False) as app_wrap:
+        chat = gr.Chatbot(label="", show_label=False, height="64vh")
+        with gr.Row():
+            uploads = gr.Files(
+                label="Upload docs/images (PDF, DOCX, CSV, PNG, JPG)",
+                file_types=["file"], file_count="multiple", height=68
             )
-        )
-
-    with gr.Row():
-        uploads = gr.Files(
-            label="Upload docs/images (PDF, DOCX, CSV, PNG, JPG)",
-            file_types=["file"], file_count="multiple", height=68
-        )
-
-    with gr.Row():
-        msg = gr.Textbox(
-            label="",
-            show_label=False,
-            placeholder="Paste your scenario here (attach files below). ClarityOps will ask clarifications first.",
-            scale=10
-        )
-        send = gr.Button("Send", scale=1)
-        clear = gr.Button("Clear chat", scale=1)
+        with gr.Row():
+            msg = gr.Textbox(
+                label="",
+                show_label=False,
+                placeholder="Continue here. Paste scenario details, add files below.",
+                scale=10
+            )
+            send = gr.Button("Send", scale=1)
+            clear = gr.Button("Clear chat", scale=1)
 
+    # ---- State
     state_history = gr.State(value=[])
     state_uploaded = gr.State(value=[])
     state_awaiting = gr.State(value=False)  # False -> Phase 1 next; True -> awaiting answers for Phase 2
 
-    def _store_uploads_and_hide(files, current):
+    # ---- Uploads
+    def _store_uploads(files, current):
         paths = []
         for f in (files or []):
             paths.append(getattr(f, "name", None) or f)
-        overlay_hidden = gr.update(value='<div id="handshake-overlay" class="hidden"></div>')
-        return (current or []) + paths, overlay_hidden
+        return (current or []) + paths
 
-    uploads.change(fn=_store_uploads_and_hide,
-                   inputs=[uploads, state_uploaded],
-                   outputs=[state_uploaded, handshake])
+    uploads.change(fn=_store_uploads, inputs=[uploads, state_uploaded], outputs=state_uploaded)
 
+    # ---- Core send (used by both hero input and chat input)
     def _on_send(user_msg, history, up_paths, awaiting):
-        hide_overlay = gr.update(value='<div id="handshake-overlay" class="hidden"></div>')
         try:
             if not user_msg or not user_msg.strip():
-                return history, "", history, awaiting, hide_overlay
+                # no toggle on empty
+                return history, "", history, awaiting
             new_history, new_awaiting = clarityops_reply(
                 user_msg.strip(), history or [], None, up_paths or [], awaiting_answers=awaiting
             )
-            return new_history, "", new_history, new_awaiting, hide_overlay
+            return new_history, "", new_history, new_awaiting
         except Exception as e:
             err = f"Error: {e}"
             try: traceback.print_exc()
             except Exception: pass
             new_hist = (history or []) + [(user_msg or "", err)]
-            return new_hist, "", new_hist, awaiting, hide_overlay
+            return new_hist, "", new_hist, awaiting
+
+    # ---- Hero -> App transition + first send
+    def _hero_start(user_msg, history, up_paths, awaiting):
+        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
+        )
 
+    hero_send.click(
+        _hero_start,
+        inputs=[hero_msg, state_history, state_uploaded, state_awaiting],
+        outputs=[chat, msg, state_history, state_awaiting, hero_wrap, app_wrap, hero_msg],
+        concurrency_limit=2, queue=True
+    )
+    hero_msg.submit(
+        _hero_start,
+        inputs=[hero_msg, state_history, state_uploaded, state_awaiting],
+        outputs=[chat, msg, state_history, state_awaiting, hero_wrap, app_wrap, hero_msg],
+        concurrency_limit=2, queue=True
+    )
+
+    # ---- Normal chat interactions after hero is gone
     send.click(_on_send, inputs=[msg, state_history, state_uploaded, state_awaiting],
-               outputs=[chat, msg, state_history, state_awaiting, handshake],
+               outputs=[chat, msg, state_history, state_awaiting],
                concurrency_limit=2, queue=True)
-
     msg.submit(_on_send, inputs=[msg, state_history, state_uploaded, state_awaiting],
-               outputs=[chat, msg, state_history, state_awaiting, handshake],
+               outputs=[chat, msg, state_history, state_awaiting],
                concurrency_limit=2, queue=True)
 
     def _on_clear():
-        # Reset everything, show handshake again
-        return [], "", [], False, (
-            '<div id="handshake-overlay"><div class="panel">'
-            'ClarityOps loaded. Paste your scenario and attach files. '
-            'I’ll ask up to 5 clarifications, then produce the structured analysis'
-            '</div></div>'
+        # reset to fresh hero screen
+        return (
+            [], "", [], False,
+            gr.update(visible=True),   # show hero
+            gr.update(visible=False),  # hide app
+            ""                         # clear hero input
         )
 
-    clear.click(_on_clear, None, [chat, msg, state_history, state_awaiting, handshake])
+    clear.click(_on_clear, None, [chat, msg, state_history, state_awaiting, hero_wrap, app_wrap, hero_msg])
 
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", "7860"))