Update app.py

app.py

@@ -1,16 +1,14 @@
 """
 Sloth — AI Workforce Intelligence Demo
-Hugging Face Spaces deployment
+Hugging Face Spaces · Gradio 6 compatible
 
-Works in two modes:
-  1. STANDALONE — if no SLOTH_BACKEND_URL is set, uses built-in demo data (always live)
-  2. CONNECTED  — if SLOTH_BACKEND_URL is set, calls your private backend (real engine)
+Standalone mode: works immediately with built-in demo data (no backend needed)
+Connected mode:  set SLOTH_BACKEND_URL secret to proxy to your private engine
 """
 
 import gradio as gr
 import requests
 import os
-import json
 import datetime
 import random
 
@@ -18,7 +16,6 @@ BACKEND_URL = os.environ.get("SLOTH_BACKEND_URL", "").strip()
 API_TOKEN   = os.environ.get("SLOTH_API_TOKEN", "sloth-demo-token-2025")
 HEADERS     = {"Authorization": f"Bearer {API_TOKEN}", "Content-Type": "application/json"}
 
-# ── Built-in demo data (used when no backend is connected) ─────────────────
 DEMO_DB = {
     "malaysia": {
         "country": "Malaysia", "flag": "🇲🇾",
@@ -36,8 +33,8 @@ DEMO_DB = {
         "country": "Germany", "flag": "🇩🇪",
         "min_wage": {"amount": "EUR 12.41/hour", "effective": "2024-01-01", "status": "current"},
         "tax": {"corporate": "15% + solidarity", "personal_top": "45%", "last_updated": "2024-01-01"},
-        "social_security": {"employer_total": "~20%", "employee_total": "~20%", "health": "14.6%"},
-        "contractor_rules": "Strict bogus self-employment (Scheinselbständigkeit) laws. High misclassification risk.",
+        "social_security": {"employer_total": "~20%", "employee_total": "~20%", "health_insurance": "14.6%"},
+        "contractor_rules": "Strict Scheinselbständigkeit laws. High misclassification risk.",
         "recent_changes": [
             {"date": "2024-01-01", "type": "Payroll", "summary": "Minimum wage increased to EUR 12.41/hour", "severity": "high"},
             {"date": "2024-02-15", "type": "Compliance", "summary": "New Works Council AI consultation requirements", "severity": "medium"},
@@ -49,10 +46,10 @@ DEMO_DB = {
         "min_wage": {"amount": "BRL 1,412/month", "effective": "2024-01-01", "status": "current"},
         "tax": {"corporate": "15-25%", "personal_top": "27.5%", "last_updated": "2024-01-01"},
         "social_security": {"inss_employer": "20%", "inss_employee": "7.5-14%", "fgts": "8%"},
-        "contractor_rules": "CLT employment strongly preferred by courts. PJ contractor arrangements under scrutiny.",
+        "contractor_rules": "CLT employment strongly preferred by courts. PJ arrangements under scrutiny.",
         "recent_changes": [
             {"date": "2024-01-01", "type": "Payroll", "summary": "Minimum wage raised to BRL 1,412", "severity": "high"},
-            {"date": "2024-02-01", "type": "Compliance", "summary": "Updated eSocial reporting v3.1 mandatory", "severity": "high"},
+            {"date": "2024-02-01", "type": "Compliance", "summary": "eSocial reporting v3.1 now mandatory", "severity": "high"},
             {"date": "2024-03-10", "type": "Tax", "summary": "New Simples Nacional thresholds for 2024", "severity": "medium"},
         ],
         "expansion_score": 61, "eor_available": True,
@@ -62,7 +59,7 @@ DEMO_DB = {
         "min_wage": {"amount": "No statutory minimum wage", "effective": "N/A", "status": "info"},
         "tax": {"corporate": "17%", "personal_top": "24%", "last_updated": "2024-04-01"},
         "social_security": {"cpf_employer": "17%", "cpf_employee": "20%", "applies_to": "Citizens/PRs only"},
-        "contractor_rules": "Clear contractor framework. Employment Act amended 2021 covers most workers.",
+        "contractor_rules": "Clear framework. Employment Act 2021 covers most workers.",
         "recent_changes": [
             {"date": "2024-04-01", "type": "Tax", "summary": "GST increased to 9% from 8%", "severity": "medium"},
             {"date": "2024-01-01", "type": "Compliance", "summary": "Flexible work arrangement guidelines effective", "severity": "low"},
@@ -74,7 +71,7 @@ DEMO_DB = {
         "min_wage": {"amount": "USD 7.25/hour federal (state rates vary)", "effective": "2009-07-24", "status": "outdated"},
         "tax": {"corporate": "21%", "personal_top": "37%", "last_updated": "2024-01-01"},
         "social_security": {"fica_employer": "7.65%", "fica_employee": "7.65%", "futa": "0.6%"},
-        "contractor_rules": "ABC test or economic realities test varies by state. California AB5 most restrictive.",
+        "contractor_rules": "ABC test varies by state. California AB5 most restrictive.",
         "recent_changes": [
             {"date": "2024-01-01", "type": "Compliance", "summary": "Corporate Transparency Act BOI reporting effective", "severity": "high"},
             {"date": "2024-04-23", "type": "Labor", "summary": "DOL new overtime threshold rule finalized", "severity": "high"},
@@ -83,13 +80,13 @@ DEMO_DB = {
     },
     "india": {
         "country": "India", "flag": "🇮🇳",
-        "min_wage": {"amount": "INR 176–423/day (varies by state & skill)", "effective": "2024-04-01", "status": "current"},
+        "min_wage": {"amount": "INR 176-423/day (varies by state & skill)", "effective": "2024-04-01", "status": "current"},
         "tax": {"corporate": "22% (domestic)", "personal_top": "30%", "last_updated": "2024-04-01"},
         "social_security": {"epf_employer": "12%", "epf_employee": "12%", "esi": "3.25% employer"},
-        "contractor_rules": "Contract Labour Act regulates use of contractors. Principal employer liability applies.",
+        "contractor_rules": "Contract Labour Act regulates use. Principal employer liability applies.",
         "recent_changes": [
-            {"date": "2024-04-01", "type": "Tax", "summary": "New TDS provisions for online gaming / crypto gains", "severity": "medium"},
-            {"date": "2024-01-15", "type": "Compliance", "summary": "EPFO UAN autogeneration now mandatory for new hires", "severity": "medium"},
+            {"date": "2024-04-01", "type": "Tax", "summary": "New TDS provisions for online gaming / crypto", "severity": "medium"},
+            {"date": "2024-01-15", "type": "Compliance", "summary": "EPFO UAN autogeneration mandatory for new hires", "severity": "medium"},
         ],
         "expansion_score": 68, "eor_available": True,
     },
@@ -98,10 +95,9 @@ DEMO_DB = {
         "min_wage": {"amount": "JPY 1,004/hour (national avg)", "effective": "2023-10-01", "status": "current"},
         "tax": {"corporate": "23.2%", "personal_top": "45%", "last_updated": "2024-01-01"},
         "social_security": {"health_employer": "5.0%", "pension_employer": "9.15%", "employment_insurance": "0.95%"},
-        "contractor_rules": "Worker dispatch laws strictly regulate contractor use. Misclassification fines are heavy.",
+        "contractor_rules": "Worker dispatch laws strictly regulate contractor use. Misclassification fines heavy.",
         "recent_changes": [
             {"date": "2024-04-01", "type": "Labor", "summary": "Overtime disclosure rules for large firms effective", "severity": "medium"},
-            {"date": "2024-01-01", "type": "Compliance", "summary": "Childcare leave promotion requirements strengthened", "severity": "low"},
         ],
         "expansion_score": 78, "eor_available": True,
     },
@@ -110,7 +106,7 @@ DEMO_DB = {
         "min_wage": {"amount": "AUD 23.23/hour", "effective": "2023-07-01", "status": "current"},
         "tax": {"corporate": "30% (25% for small business)", "personal_top": "45%", "last_updated": "2024-07-01"},
         "social_security": {"superannuation_employer": "11%", "medicare_levy": "2%"},
-        "contractor_rules": "High misclassification risk post-2022 High Court rulings. Substance over form test applies.",
+        "contractor_rules": "High misclassification risk post-2022 High Court rulings. Substance over form test.",
         "recent_changes": [
             {"date": "2024-07-01", "type": "Payroll", "summary": "Superannuation rate increases to 11.5%", "severity": "high"},
             {"date": "2024-01-01", "type": "Labor", "summary": "Right to disconnect legislation effective", "severity": "medium"},
@@ -135,13 +131,12 @@ RATING_ICON   = {"High": "🟢", "Medium": "🟡", "Low": "🔴"}
 
 
 def query_standalone(query: str) -> dict:
-    """Query the built-in demo database."""
     raw = query.lower()
     for key, data in DEMO_DB.items():
         if key in raw:
             return {
                 "status": "success",
-                "queried_at": datetime.datetime.utcnow().isoformat() + "Z",
+                "queried_at": datetime.datetime.now(datetime.timezone.utc).isoformat().replace("+00:00", "Z"),
                 "source": "Sloth Intelligence Engine v1.0-MVP (Demo)",
                 "country": data["country"], "flag": data["flag"],
                 "summary": {
@@ -161,45 +156,31 @@ def query_standalone(query: str) -> dict:
             }
     return {
         "status": "no_match",
-        "message": f"No data found for: '{query}'",
+        "message": f"No data for: '{query}'",
         "available_countries": [v["country"] for v in DEMO_DB.values()],
     }
 
 
-def call_backend(query: str) -> dict:
-    """Call the connected private backend."""
-    try:
-        resp = requests.post(
-            f"{BACKEND_URL}/query",
-            headers=HEADERS,
-            json={"query": query},
-            timeout=15,
-        )
-        resp.raise_for_status()
-        return resp.json()
-    except requests.exceptions.ConnectionError:
-        return {"status": "error", "message": "⚠ Backend unreachable. Falling back to demo data.", "fallback": True}
-    except Exception as e:
-        return {"status": "error", "message": f"⚠ Backend error: {str(e)}", "fallback": True}
-
-
 def get_data(query: str) -> dict:
     if BACKEND_URL:
-        result = call_backend(query)
-        if result.get("fallback"):
-            return query_standalone(query)
-        return result
+        try:
+            resp = requests.post(
+                f"{BACKEND_URL}/query", headers=HEADERS,
+                json={"query": query}, timeout=15,
+            )
+            resp.raise_for_status()
+            return resp.json()
+        except Exception:
+            pass
     return query_standalone(query)
 
 
 def format_result(data: dict) -> tuple:
     if data.get("status") == "error":
         return data.get("message", "Unknown error"), "", "", ""
-
     if data.get("status") == "no_match":
         avail = ", ".join(data.get("available_countries", []))
-        return f"**No match for that query.**\n\nCurrently monitoring: {avail}", "", "", ""
-
+        return f"**No match.**  \nCurrently monitoring: {avail}", "", "", ""
     if data.get("status") != "success":
         return "Unexpected response.", "", "", ""
 
@@ -208,421 +189,128 @@ def format_result(data: dict) -> tuple:
     ts      = data.get("queried_at", "")[:19].replace("T", " ")
     alerts  = data.get("alert_count", 0)
     sources = data.get("monitored_sources", 0)
-    mode    = " · Demo mode" if "Demo" in data.get("source", "") else ""
-
-    header = f"# {flag} {country}\n*Queried {ts} UTC · {sources} sources monitored · {alerts} recent alerts{mode}*\n---"
-
-    s = data.get("summary", {})
-    mw = s.get("minimum_wage", {})
-    tx = s.get("tax_rates", {})
-    ss = s.get("social_security", {})
-    cr = s.get("contractor_classification", "N/A")
-    wage_status = {"current": "✅ Current", "outdated": "⚠ Outdated", "info": "ℹ Info"}.get(mw.get("status", ""), "")
-
-    summary = (
-        f"## 📋 Compliance Summary\n\n"
-        f"**Minimum Wage**  \n{mw.get('amount','N/A')} — effective {mw.get('effective','N/A')} {wage_status}\n\n"
-        f"**Tax Rates**  \n- Corporate: `{tx.get('corporate','N/A')}`\n- Personal (top): `{tx.get('personal_top','N/A')}`\n- Last updated: {tx.get('last_updated','N/A')}\n\n"
-        f"**Social Security**  \n" + "\n".join(f"- {k.replace('_',' ').title()}: `{v}`" for k, v in ss.items()) +
-        f"\n\n**Contractor Classification**  \n{cr}"
-    )
-
-    changes = data.get("recent_regulatory_changes", [])
-    if changes:
-        rows = "\n".join(
-            f"| {SEVERITY_ICON.get(c.get('severity','low'),'⚪')} | {c.get('date','')} | {c.get('type','')} | {c.get('summary','')} |"
-            for c in changes
-        )
-        changes_md = f"## ⚡ Recent Regulatory Changes\n\n| Sev | Date | Type | Summary |\n|---|---|---|---|\n{rows}"
-    else:
-        changes_md = "## ⚡ Recent Regulatory Changes\n\nNo recent changes detected."
-
-    er = data.get("expansion_readiness", {})
-    score  = er.get("score", 0)
-    rating = er.get("rating", "N/A")
-    eor    = "✅ Available" if er.get("eor_available") else "❌ Not available"
-    bar    = "█" * (score // 10) + "░" * (10 - score // 10)
-    expansion = (
-        f"## 🌍 Expansion Readiness\n\n"
-        f"**Score: {score}/100** {RATING_ICON.get(rating,'⚪')} {rating}\n\n"
-        f"`{bar}` {score}%\n\n**EOR:** {eor}"
-    )
-
-    return header, summary, changes_md, expansion
-
-
-def run_query(query: str):
-    if not query.strip():
-        return "Enter a country or compliance question above.", "", "", ""
-    return format_result(get_data(query))
-
-
-CSS = """
-body, .gradio-container { font-family: 'Inter', system-ui, sans-serif !important; background: #F8FAFC !important; }
-#header-banner { background: linear-gradient(135deg, #0B1F3A 0%, #12294D 60%, #0D9488 100%); border-radius: 12px; padding: 28px 32px 20px; margin-bottom: 16px; }
-#header-banner h1 { font-size: 2.4rem; font-weight: 700; letter-spacing: 0.1em; color: white !important; margin: 0 0 4px; }
-#header-banner p { color: #5DCAA5; font-size: 1rem; margin: 0; }
-textarea, input[type=text] { border: 1.5px solid #CBD5E1 !important; border-radius: 8px !important; background: white !important; }
-textarea:focus, input[type=text]:focus { border-color: #0D9488 !important; box-shadow: 0 0 0 3px rgba(13,148,136,0.15) !important; }
-button.primary { background: #0D9488 !important; border: none !important; border-radius: 8px !important; font-weight: 600 !important; color: white !important; }
-button.primary:hover { background: #0F6E56 !important; }
-.result-panel { background: white; border: 1px solid #E2E8F0; border-radius: 10px; padding: 20px; margin-top: 12px; }
-.markdown-body h1 { color: #0B1F3A !important; font-size: 1.5rem !important; }
-.markdown-body h2 { color: #0D9488 !important; font-size: 1.15rem !important; border-bottom: 1px solid #E2E8F0; padding-bottom: 4px; }
-.markdown-body code { background: #F0FDFA !important; color: #0B1F3A !important; border-radius: 4px; padding: 1px 5px; }
-.markdown-body table { border-collapse: collapse; width: 100%; }
-.markdown-body td, .markdown-body th { border: 1px solid #E2E8F0 !important; padding: 6px 12px !important; font-size: 0.9rem; }
-.markdown-body th { background: #F8FAFC !important; }
-"""
-
-HEADER_HTML = """
-<div id="header-banner">
-  <h1>🦥 SLOTH</h1>
-  <p>AI Workforce Intelligence · Real-time compliance monitoring across 150+ countries</p>
-</div>
-"""
-
-with gr.Blocks(css=CSS, title="Sloth — AI Workforce Intelligence") as demo:
-    gr.HTML(HEADER_HTML)
-
-    with gr.Row():
-        with gr.Column(scale=5):
-            query_box = gr.Textbox(
-                placeholder="Try: 'Malaysia payroll compliance' or 'Germany minimum wage changes'",
-                label="Ask Sloth", lines=2, max_lines=4,
-            )
-        with gr.Column(scale=1, min_width=120):
-            gr.Markdown(" ")
-            submit_btn = gr.Button("Run Query →", variant="primary")
-
-    gr.Markdown("<div style='color:#64748B;font-size:0.85rem;margin-bottom:6px'>Quick examples:</div>")
-    gr.Examples(examples=EXAMPLE_QUERIES, inputs=query_box, label="")
-
-    with gr.Row():
-        header_out = gr.Markdown(label="", elem_classes=["result-panel"])
-    with gr.Row():
-        with gr.Column(scale=1):
-            summary_out = gr.Markdown(label="", elem_classes=["result-panel"])
-        with gr.Column(scale=1):
-            changes_out = gr.Markdown(label="", elem_classes=["result-panel"])
-    with gr.Row():
-        expansion_out = gr.Markdown(label="", elem_classes=["result-panel"])
-
-    gr.Markdown(
-        "<br><center style='color:#94A3B8;font-size:0.8rem'>"
-        "Sloth MVP · 8 countries in demo · Full 150-country coverage in production · "
-        "<a href='mailto:hello@sloth.ai' style='color:#0D9488'>hello@sloth.ai</a>"
-        "</center>"
-    )
-
-    submit_btn.click(fn=run_query, inputs=query_box, outputs=[header_out, summary_out, changes_out, expansion_out])
-    query_box.submit(fn=run_query, inputs=query_box, outputs=[header_out, summary_out, changes_out, expansion_out])
+    mode    = " · Demo" if "Demo" in data.get("source", "") else ""
 
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
-
-HEADERS = {
-    "Authorization": f"Bearer {API_TOKEN}",
-    "Content-Type":  "application/json",
-}
-
-EXAMPLE_QUERIES = [
-    "What are Malaysia's payroll compliance requirements?",
-    "Show me Germany's minimum wage and recent changes",
-    "Brazil contractor classification rules",
-    "Singapore expansion readiness report",
-    "United States overtime compliance 2024",
-    "India EPF and social security rates",
-]
-
-SEVERITY_ICON = {"high": "🔴", "medium": "🟡", "low": "🟢"}
-RATING_ICON   = {"High": "🟢", "Medium": "🟡", "Low": "🔴"}
-
-
-# ── API call ───────────────────────────────────────────────────────────────
-
-def call_backend(query: str) -> dict:
-    try:
-        resp = requests.post(
-            f"{BACKEND_URL}/query",
-            headers=HEADERS,
-            json={"query": query},
-            timeout=15,
-        )
-        resp.raise_for_status()
-        return resp.json()
-    except requests.exceptions.ConnectionError:
-        return {"status": "error", "message": "⚠ Backend offline. Start your local server and set SLOTH_BACKEND_URL."}
-    except requests.exceptions.Timeout:
-        return {"status": "error", "message": "⚠ Request timed out. Check your backend URL."}
-    except Exception as e:
-        return {"status": "error", "message": f"⚠ Error: {str(e)}"}
-
-
-# ── Result formatter ───────────────────────────────────────────────────────
-
-def format_result(data: dict) -> tuple[str, str, str, str]:
-    """Returns (header_md, summary_md, changes_md, expansion_md)"""
-
-    if data.get("status") == "error":
-        err = data.get("message", "Unknown error")
-        return err, "", "", ""
-
-    if data.get("status") == "no_match":
-        msg = data.get("message", "No data found.")
-        avail = ", ".join(c.title() for c in data.get("available_countries", []))
-        return f"**{msg}**\n\nAvailable: {avail}", "", "", ""
-
-    if data.get("status") != "success":
-        return "Unexpected response from backend.", "", "", ""
-
-    flag    = data.get("flag", "")
-    country = data.get("country", "")
-    ts      = data.get("queried_at", "")[:19].replace("T", " ")
-    alerts  = data.get("alert_count", 0)
-    sources = data.get("monitored_sources", 0)
-
-    # ── Header ──
     header = (
         f"# {flag} {country}\n"
-        f"*Queried {ts} UTC · {sources} sources monitored · {alerts} recent alerts*\n"
-        f"---"
+        f"*Queried {ts} UTC · {sources} sources monitored · {alerts} recent alerts{mode}*\n\n---"
     )
 
-    # ── Summary ──
-    s = data.get("summary", {})
-    mw = s.get("minimum_wage", {})
-    tx = s.get("tax_rates", {})
-    ss = s.get("social_security", {})
-    cr = s.get("contractor_classification", "N/A")
-
-    wage_status = {"current": "✅ Current", "outdated": "⚠ Outdated", "info": "ℹ Info"}.get(mw.get("status", ""), "")
+    s   = data.get("summary", {})
+    mw  = s.get("minimum_wage", {})
+    tx  = s.get("tax_rates", {})
+    ss  = s.get("social_security", {})
+    cr  = s.get("contractor_classification", "N/A")
+    ws  = {"current": "✅ Current", "outdated": "⚠ Outdated", "info": "ℹ Info"}.get(mw.get("status", ""), "")
 
     summary = (
-        f"## 📋 Compliance Summary\n\n"
-        f"**Minimum Wage**  \n"
-        f"{mw.get('amount','N/A')} — effective {mw.get('effective','N/A')} {wage_status}\n\n"
-        f"**Tax Rates**  \n"
-        f"- Corporate: `{tx.get('corporate','N/A')}`\n"
-        f"- Personal (top): `{tx.get('personal_top','N/A')}`\n"
+        "## 📋 Compliance Summary\n\n"
+        f"**Minimum Wage**  \n{mw.get('amount','N/A')} — effective {mw.get('effective','N/A')} {ws}\n\n"
+        "**Tax Rates**  \n"
+        f"- Corporate: `{tx.get('corporate','N/A')}`  \n"
+        f"- Personal (top rate): `{tx.get('personal_top','N/A')}`  \n"
         f"- Last updated: {tx.get('last_updated','N/A')}\n\n"
-        f"**Social Security**  \n"
-        + "\n".join(f"- {k.replace('_',' ').title()}: `{v}`" for k, v in ss.items()) +
-        f"\n\n**Contractor Classification**  \n{cr}"
+        "**Social Security**  \n"
+        + "\n".join(f"- {k.replace('_',' ').title()}: `{v}`" for k, v in ss.items())
+        + f"\n\n**Contractor Classification**  \n{cr}"
     )
 
-    # ── Recent changes ──
     changes = data.get("recent_regulatory_changes", [])
     if changes:
         rows = "\n".join(
-            f"| {SEVERITY_ICON.get(c.get('severity','low'), '⚪')} {c.get('date','')} "
+            f"| {SEVERITY_ICON.get(c.get('severity','low'),'⚪')} | {c.get('date','')} "
             f"| {c.get('type','')} | {c.get('summary','')} |"
             for c in changes
         )
         changes_md = (
-            f"## ⚡ Recent Regulatory Changes\n\n"
-            f"| Severity | Date | Type | Summary |\n"
-            f"|---|---|---|---|\n"
-            f"{rows}"
+            "## ⚡ Recent Regulatory Changes\n\n"
+            "| Sev | Date | Type | Summary |\n|---|---|---|---|\n" + rows
         )
     else:
         changes_md = "## ⚡ Recent Regulatory Changes\n\nNo recent changes detected."
 
-    # ── Expansion readiness ──
-    er = data.get("expansion_readiness", {})
+    er     = data.get("expansion_readiness", {})
     score  = er.get("score", 0)
     rating = er.get("rating", "N/A")
     eor    = "✅ Available" if er.get("eor_available") else "❌ Not available"
     bar    = "█" * (score // 10) + "░" * (10 - score // 10)
 
     expansion = (
-        f"## 🌍 Expansion Readiness\n\n"
+        "## 🌍 Expansion Readiness\n\n"
         f"**Score: {score}/100** {RATING_ICON.get(rating,'⚪')} {rating}\n\n"
         f"`{bar}` {score}%\n\n"
-        f"**EOR (Employer of Record):** {eor}"
+        f"**Employer of Record (EOR):** {eor}"
     )
-
     return header, summary, changes_md, expansion
 
 
-# ── Main query function ────────────────────────────────────────────────────
-
 def run_query(query: str):
     if not query.strip():
         return "Enter a country or compliance question above.", "", "", ""
+    return format_result(get_data(query))
 
-    data = call_backend(query)
-    return format_result(data)
-
-
-# ── CSS theme — navy/teal matching pitch deck ──────────────────────────────
 
 CSS = """
-:root {
-    --navy:   #0B1F3A;
-    --teal:   #0D9488;
-    --light:  #F0FDFA;
-}
-
-body, .gradio-container {
-    font-family: 'Inter', system-ui, sans-serif !important;
-    background: #F8FAFC !important;
-}
-
-.gradio-container > .main > .wrap {
-    max-width: 900px;
-    margin: 0 auto;
-}
-
-/* Header banner */
 #header-banner {
     background: linear-gradient(135deg, #0B1F3A 0%, #12294D 60%, #0D9488 100%);
-    border-radius: 12px;
-    padding: 28px 32px 20px;
-    margin-bottom: 16px;
-    color: white;
+    border-radius: 12px; padding: 28px 32px 20px; margin-bottom: 16px;
 }
-
 #header-banner h1 {
-    font-size: 2.4rem;
-    font-weight: 700;
-    letter-spacing: 0.1em;
-    color: white !important;
-    margin: 0 0 4px;
-}
-
-#header-banner p {
-    color: #5DCAA5;
-    font-size: 1rem;
-    margin: 0;
-}
-
-/* Search box */
-textarea, input[type=text] {
-    border: 1.5px solid #CBD5E1 !important;
-    border-radius: 8px !important;
-    font-size: 1rem !important;
-    background: white !important;
-}
-
-textarea:focus, input[type=text]:focus {
-    border-color: #0D9488 !important;
-    box-shadow: 0 0 0 3px rgba(13,148,136,0.15) !important;
-}
-
-/* Buttons */
-button.primary {
-    background: #0D9488 !important;
-    border: none !important;
-    border-radius: 8px !important;
-    font-weight: 600 !important;
-    font-size: 1rem !important;
-    color: white !important;
-    padding: 10px 28px !important;
-    cursor: pointer !important;
-    transition: background 0.2s !important;
-}
-
-button.primary:hover {
-    background: #0F6E56 !important;
-}
-
-button.secondary {
-    background: transparent !important;
-    border: 1.5px solid #CBD5E1 !important;
-    border-radius: 8px !important;
-    color: #64748B !important;
-    font-size: 0.9rem !important;
-    cursor: pointer !important;
+    font-size: 2.4rem; font-weight: 700; letter-spacing: 0.1em;
+    color: white !important; margin: 0 0 4px;
 }
-
-/* Result panels */
+#header-banner p { color: #5DCAA5; font-size: 1rem; margin: 0; }
 .result-panel {
-    background: white;
-    border: 1px solid #E2E8F0;
-    border-radius: 10px;
-    padding: 20px;
-    margin-top: 12px;
-    box-shadow: 0 1px 3px rgba(0,0,0,0.06);
+    background: white; border: 1px solid #E2E8F0;
+    border-radius: 10px; padding: 20px; margin-top: 12px;
 }
-
-/* Markdown output */
-.markdown-body h1 { color: #0B1F3A !important; font-size: 1.5rem !important; }
-.markdown-body h2 { color: #0D9488 !important; font-size: 1.15rem !important; border-bottom: 1px solid #E2E8F0; padding-bottom: 4px; }
-.markdown-body code { background: #F0FDFA !important; color: #0B1F3A !important; border-radius: 4px; padding: 1px 5px; }
-.markdown-body table { border-collapse: collapse; width: 100%; }
-.markdown-body td, .markdown-body th { border: 1px solid #E2E8F0 !important; padding: 6px 12px !important; font-size: 0.9rem; }
-.markdown-body th { background: #F8FAFC !important; }
-
-/* Example pills */
-.example-label { color: #64748B; font-size: 0.85rem; margin-bottom: 6px; }
 """
 
 HEADER_HTML = """
 <div id="header-banner">
   <h1>🦥 SLOTH</h1>
-  <p>AI Workforce Intelligence · Real-time compliance monitoring across 150+ countries</p>
+  <p>AI Workforce Intelligence &nbsp;·&nbsp; Real-time compliance monitoring across 150+ countries</p>
 </div>
 """
 
-# ── Build Gradio UI ────────────────────────────────────────────────────────
-
-with gr.Blocks(css=CSS, title="Sloth — AI Workforce Intelligence") as demo:
-
+with gr.Blocks(title="Sloth — AI Workforce Intelligence") as demo:
     gr.HTML(HEADER_HTML)
 
     with gr.Row():
         with gr.Column(scale=5):
             query_box = gr.Textbox(
                 placeholder="Try: 'Malaysia payroll compliance' or 'Germany minimum wage changes'",
-                label="Ask Sloth",
-                lines=2,
-                max_lines=4,
+                label="Ask Sloth", lines=2, max_lines=4,
             )
-        with gr.Column(scale=1, min_width=120):
-            gr.Markdown(" ")
+        with gr.Column(scale=1, min_width=130):
+            gr.Markdown("&nbsp;")
             submit_btn = gr.Button("Run Query →", variant="primary")
 
-    # Example queries
-    gr.Markdown("<div class='example-label'>Quick examples:</div>")
-    gr.Examples(
-        examples=EXAMPLE_QUERIES,
-        inputs=query_box,
-        label="",
-    )
+    gr.Markdown("<span style='color:#64748B;font-size:0.85rem'>Quick examples — click to load:</span>")
+    gr.Examples(examples=EXAMPLE_QUERIES, inputs=query_box, label="")
 
-    # Output panels
     with gr.Row():
-        header_out = gr.Markdown(label="", elem_classes=["result-panel"])
-
+        header_out = gr.Markdown(elem_classes=["result-panel"])
     with gr.Row():
         with gr.Column(scale=1):
-            summary_out = gr.Markdown(label="", elem_classes=["result-panel"])
+            summary_out = gr.Markdown(elem_classes=["result-panel"])
         with gr.Column(scale=1):
-            changes_out = gr.Markdown(label="", elem_classes=["result-panel"])
-
+            changes_out = gr.Markdown(elem_classes=["result-panel"])
     with gr.Row():
-        expansion_out = gr.Markdown(label="", elem_classes=["result-panel"])
+        expansion_out = gr.Markdown(elem_classes=["result-panel"])
 
-    # Footer
     gr.Markdown(
-        "<br><center style='color:#94A3B8;font-size:0.8rem'>"
-        "Sloth MVP · Data sourced from 5,000+ government portals, labor law sites, and talent platforms · "
-        "<a href='https://sloth.ai' style='color:#0D9488'>sloth.ai</a>"
+        "<center style='color:#94A3B8;font-size:0.8rem;margin-top:16px'>"
+        "Sloth MVP &nbsp;·&nbsp; 8 countries in demo &nbsp;·&nbsp; "
+        "Full 150-country coverage in production &nbsp;·&nbsp; "
+        "<a href='mailto:hello@sloth.ai' style='color:#0D9488'>hello@sloth.ai</a>"
         "</center>"
     )
 
-    # Wire up
-    submit_btn.click(
-        fn=run_query,
-        inputs=query_box,
-        outputs=[header_out, summary_out, changes_out, expansion_out],
-    )
-    query_box.submit(
-        fn=run_query,
-        inputs=query_box,
-        outputs=[header_out, summary_out, changes_out, expansion_out],
-    )
+    submit_btn.click(fn=run_query, inputs=query_box,
+                     outputs=[header_out, summary_out, changes_out, expansion_out])
+    query_box.submit(fn=run_query, inputs=query_box,
+                     outputs=[header_out, summary_out, changes_out, expansion_out])
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
+    demo.launch(server_name="0.0.0.0", server_port=7860, css=CSS, ssr_mode=False)