Create frontend/app.py

frontend/app.py

@@ -0,0 +1,37 @@
+import chainlit as cl
+import httpx
+
+# URL do nosso backend FastAPI
+API_URL = "http://localhost:8000/chat"
+
+@cl.on_chat_start
+def start():
+    cl.user_session.set("history", [])
+
+@cl.on_message
+async def main(message: cl.Message):
+    # Cria uma mensagem de resposta vazia para preencher com o stream
+    msg = cl.Message(content="")
+    await msg.send()
+
+    try:
+        # Usamos httpx para fazer uma requisição de streaming para o nosso backend
+        async with httpx.AsyncClient(timeout=60) as client:
+            # Fazemos um POST para o endpoint /chat com a mensagem do usuário
+            async with client.stream("POST", API_URL, json={"content": message.content}) as response:
+                # Se a requisição falhar, mostra um erro
+                if response.status_code != 200:
+                    await msg.update(content=f"Erro: {response.text}")
+                    return
+
+                # Itera sobre os pedaços (chunks) de texto recebidos do backend
+                async for chunk in response.aiter_text():
+                    await msg.stream_token(chunk)
+        
+        # Finaliza a mensagem
+        await msg.update()
+
+    except httpx.ConnectError:
+        await msg.update(content="Erro de conexão: Não foi possível se conectar ao backend. Verifique se ele está rodando.")
+    except Exception as e:
+        await msg.update(content=f"Ocorreu um erro: {e}")