Update app.py

app.py

@@ -732,6 +732,53 @@ def render_error(msg: str) -> str:
 """
 
 
+
+def render_evidence_summary(results: list[dict]) -> str:
+    """Render a compact, visible summary of the retrieved RAG evidence."""
+    if not results:
+        return """
+<div style="font-family:'IBM Plex Mono',monospace;background:#0f172a;color:#94a3b8;
+            padding:1rem;border-radius:10px;border:1px solid #1e3a5f;margin-top:0.8rem">
+  No retrieved evidence to display.
+</div>
+"""
+
+    rows = []
+    for i, r in enumerate(results, 1):
+        source_id = esc(r.get("source_id", "unknown"))
+        authority = esc(r.get("authority", "?"))
+        article = esc(r.get("article_hint", "—") or "—")
+        norm = esc(r.get("normative_reference_hint", "") or "")
+        score = f"{float(r.get('_final', 0.0)):.3f}"
+        text = esc(str(r.get("text", ""))[:260].strip())
+        norm_line = f" · {norm}" if norm else ""
+        rows.append(f"""
+  <div style="background:#111827;border:1px solid #26364f;border-radius:8px;
+              padding:0.85rem;margin:0.55rem 0">
+    <div style="display:flex;gap:0.5rem;align-items:center;flex-wrap:wrap;margin-bottom:0.35rem">
+      <span style="color:#e2e8f0;font-weight:700">SOURCE {i}</span>
+      <span style="background:#1e3a5f;color:#93c5fd;padding:2px 8px;border-radius:4px;font-size:0.72rem">{source_id}</span>
+      <span style="color:#94a3b8;font-size:0.72rem">{authority} · {article}{norm_line} · score {score}</span>
+    </div>
+    <div style="color:#cbd5e1;font-size:0.78rem;line-height:1.55">{text}...</div>
+  </div>
+""")
+
+    return f"""
+<div style="font-family:'IBM Plex Mono',monospace;background:#0f172a;color:#e2e8f0;
+            padding:1rem;border-radius:12px;border:1px solid #1e3a5f;margin:1rem 0">
+  <div style="color:#94a3b8;font-size:0.75rem;text-transform:uppercase;
+              letter-spacing:0.1em;margin-bottom:0.4rem">
+    Evidence retrieved from RAG
+  </div>
+  <div style="color:#64748b;font-size:0.74rem;margin-bottom:0.5rem">
+    Compact view of the chunks used as context for Claude. The full raw context remains available below.
+  </div>
+  {''.join(rows)}
+</div>
+"""
+
+
 # ============================================================
 # Gradio app
 # ============================================================
@@ -747,9 +794,9 @@ EXAMPLES = [
 ]
 
 
-def analyze(query: str) -> tuple[str, str]:
+def analyze(query: str) -> tuple[str, str, str]:
     if not query or not query.strip():
-        return render_error("Please enter a compliance query."), ""
+        return render_error("Please enter a compliance query."), "", ""
 
     query = query.strip()
     print("\n" + "=" * 72, flush=True)
@@ -768,12 +815,12 @@ def analyze(query: str) -> tuple[str, str]:
         )
 
     if not results:
-        return render_error("No relevant regulatory chunks found. Try rephrasing your query."), ""
+        return render_error("No relevant regulatory chunks found. Try rephrasing your query."), "", ""
 
     context = format_context(results)
     report = call_claude(query, context)
     if not report:
-        return render_error("Could not reach Claude API. Check that ANTHROPIC_API_KEY is set as a Space Secret."), context
+        return render_error("Could not reach Claude API. Check that ANTHROPIC_API_KEY is set as a Space Secret."), render_evidence_summary(results), context
 
     report = canonicalize_report(report, results)
     print(
@@ -783,7 +830,7 @@ def analyze(query: str) -> tuple[str, str]:
         flush=True,
     )
 
-    return render_report(report, query, results), context
+    return render_report(report, query, results), render_evidence_summary(results), context
 
 
 CSS = """
@@ -802,7 +849,7 @@ body { background: #0a0f1e !important; }
 .query-box textarea:focus { border-color: #38bdf8 !important; box-shadow: 0 0 0 2px rgba(56,189,248,0.15) !important; }
 .analyze-btn { background: #0369a1 !important; color: #fff !important; font-family: 'IBM Plex Mono', monospace !important; font-weight: 600 !important; letter-spacing: 0.05em !important; border: none !important; border-radius: 8px !important; height: 44px !important; font-size: 0.85rem !important; transition: background 0.2s !important; }
 .analyze-btn:hover { background: #0284c7 !important; }
-.context-box textarea { background: #0f172a !important; border: 1px solid #1e293b !important; color: #64748b !important; font-family: 'IBM Plex Mono', monospace !important; font-size: 0.75rem !important; }
+.context-box textarea { background: #0f172a !important; border: 1px solid #1e293b !important; color: #cbd5e1 !important; font-family: 'IBM Plex Mono', monospace !important; font-size: 0.75rem !important; }
 label { color: #94a3b8 !important; font-size: 0.8rem !important; }
 """
 
@@ -839,7 +886,9 @@ with gr.Blocks(css=CSS, title="RegTech BR") as demo:
 
     report_html = gr.HTML(label="Compliance Assessment")
 
-    with gr.Accordion("Retrieved regulatory context", open=False):
+    evidence_html = gr.HTML(label="Evidence retrieved from RAG")
+
+    with gr.Accordion("Full raw retrieved regulatory context", open=False):
         context_box = gr.Textbox(
             label="Raw chunks retrieved by RAG",
             lines=10,
@@ -856,8 +905,8 @@ with gr.Blocks(css=CSS, title="RegTech BR") as demo:
     """)
 
     use_example_btn.click(fn=lambda example: example or "", inputs=[example_dropdown], outputs=[query_box])
-    analyze_btn.click(fn=analyze, inputs=[query_box], outputs=[report_html, context_box])
-    query_box.submit(fn=analyze, inputs=[query_box], outputs=[report_html, context_box])
+    analyze_btn.click(fn=analyze, inputs=[query_box], outputs=[report_html, evidence_html, context_box])
+    query_box.submit(fn=analyze, inputs=[query_box], outputs=[report_html, evidence_html, context_box])
 
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 7860))