Update frontend/app.py

frontend/app.py

@@ -6,6 +6,7 @@
 
 import os
 import chainlit as cl
+from chainlit.input_widget import Switch
 import httpx
 import asyncio
 import json
@@ -200,14 +201,14 @@ def render_pipeline(fase_status: dict, detalhe: str = "", concluido: bool = Fals
 # ──────────────────────────────────────────────────────────────────────
 # STREAMING ASSÍNCRONO (usa BACKEND_API_URL)
 # ──────────────────────────────────────────────────────────────────────
-async def stream_eventos(pergunta: str, session_id: str) -> AsyncGenerator[dict, None]:
+async def stream_eventos(pergunta: str, session_id: str, deep_search: bool = False) -> AsyncGenerator[dict, None]:
     try:
         async with httpx.AsyncClient(
             timeout=httpx.Timeout(300.0, connect=5.0)
         ) as client:
             async with client.stream(
                 "POST", f"{BACKEND_API_URL}/chat",
-                json={"content": pergunta, "session_id": session_id},
+                json={"content": pergunta, "session_id": session_id, "deep_search": deep_search},
                 headers={"Accept": "application/x-ndjson", **headers},
             ) as resp:
                 if resp.status_code != 200:
@@ -730,6 +731,17 @@ async def on_download_pdf(action: cl.Action):
 
 @cl.on_chat_start
 async def start():
+    await cl.ChatSettings(
+        [
+            Switch(
+                id="busca_profunda",
+                label="🔍 Busca Profunda",
+                description="Ativa análise mais detalhada (mais lenta)",
+                initial=False,
+            )
+        ]
+    ).send()
+
     if cl.user_session.get("_welcomed"):
         return
     cl.user_session.set("_welcomed", True)
@@ -782,9 +794,15 @@ async def start():
 
 
 
+@cl.on_settings_update
+async def on_settings_update(settings: dict):
+    cl.user_session.set("busca_profunda", settings.get("busca_profunda", False))
+
+
 @cl.on_message
 async def main(message: cl.Message):
-    session_id = cl.user_session.get("id") or "default"
+    session_id    = cl.user_session.get("id") or "default"
+    busca_profunda = cl.user_session.get("busca_profunda", False)
 
     welcome_msg = cl.user_session.get("welcome_msg")
     if welcome_msg:
@@ -823,7 +841,7 @@ async def main(message: cl.Message):
         pipeline_step.input  = message.content
         pipeline_step.output = render_pipeline(fase_status, ultimo_detalhe)
 
-        async for evento in stream_eventos(message.content, session_id):
+        async for evento in stream_eventos(message.content, session_id, busca_profunda):
             tipo = evento.get("type", "")
 
             if tipo == "status":