Update frontend/app.py

frontend/app.py

@@ -1,11 +1,13 @@
-#--- START OF FILE app (25).py ---
+#--- START OF FILE app.py ---
 
 import chainlit as cl
 import httpx
 import os 
-import mimetypes 
 
 # URL do nosso backend FastAPI
+# Se estiver rodando no Docker, localhost geralmente funciona se usar network host, 
+# mas em bridge pode precisar do nome do serviço. 
+# Mantendo localhost conforme seu log de sucesso anterior (200 OK).
 API_URL = "http://localhost:8000"
 UPLOAD_URL = f"{API_URL}/upload-document"
 CHAT_URL = f"{API_URL}/chat"
@@ -40,16 +42,13 @@ async def start():
         
         try:
             # 1. Obtendo o MIME Type de forma robusta
-            # Usa file.type (padrão Chainlit) ou assume PDF
             mime_type = getattr(file, 'type', 'application/pdf') or 'application/pdf'
             
-            # 2. Lendo o conteúdo do arquivo do disco (file.path)
-            # O Chainlit armazena o arquivo em disco e fornece o caminho (file.path)
+            # 2. Lendo o conteúdo do arquivo
             with open(file.path, "rb") as f:
                 file_content = f.read()
 
-            # 3. Criando o payload para httpx. O formato é (filename, file_content_bytes, mime_type)
-            # A chave "file" deve corresponder ao argumento "file" no FastAPI (main (1).py)
+            # 3. Criando o payload
             files_payload = {
                 "file": (file.name, file_content, mime_type)
             }
@@ -58,31 +57,27 @@ async def start():
             async with httpx.AsyncClient(timeout=300) as client:
                 response = await client.post(
                     UPLOAD_URL, 
-                    files=files_payload # Note: usando o argumento 'files'
+                    files=files_payload
                 )
                 
-                # Se o upload/processamento for bem-sucedido (200 OK)
+                # CORREÇÃO AQUI: Atualiza o atributo content e depois chama update() sem argumentos
                 if response.status_code == 200:
                     RAG_INITIALIZED = True
-                    # Usa content= para forçar o argumento nomeado e evitar TypeError
-                    await msg.update(
-                        content=f"✅ Documento '{file.name}' processado com sucesso! Agora você pode fazer perguntas sobre o conteúdo do documento."
-                    )
+                    msg.content = f"✅ Documento '{file.name}' processado com sucesso! Agora você pode fazer perguntas sobre o conteúdo do documento."
+                    await msg.update()
                 else:
                     RAG_INITIALIZED = False
                     error_detail = response.json().get('detail', 'Erro desconhecido')
-                    # Usa content= para forçar o argumento nomeado e evitar TypeError
-                    await msg.update(
-                        content=f"❌ Erro ao processar o arquivo. Código: {response.status_code}. Detalhe: {error_detail}"
-                    )
+                    msg.content = f"❌ Erro ao processar o arquivo. Código: {response.status_code}. Detalhe: {error_detail}"
+                    await msg.update()
 
         except httpx.ConnectError:
-            # Usa content= para forçar o argumento nomeado e evitar TypeError
-            await msg.update(content="Erro de conexão: Não foi possível se conectar ao backend. Verifique se ele está rodando.")
+            msg.content = "Erro de conexão: Não foi possível se conectar ao backend. Verifique se ele está rodando."
+            await msg.update()
             
         except Exception as e:
-            # Usa content= para forçar o argumento nomeado e evitar TypeError
-            await msg.update(content=f"Ocorreu um erro: {e}")
+            msg.content = f"Ocorreu um erro: {e}"
+            await msg.update()
         
 @cl.on_message
 async def main(message: cl.Message):
@@ -100,35 +95,32 @@ async def main(message: cl.Message):
     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", CHAT_URL, json={"content": message.content}) as response:
-                # Se a requisição falhar, mostra um erro
                 if response.status_code != 200:
-                    # Tenta ler o erro do corpo da resposta, se disponível
                     try:
                         error_text = await response.aread()
                         error_detail = error_text.decode('utf-8')
                     except Exception:
                         error_detail = f"Status Code: {response.status_code}"
 
-                    # Usa content= para forçar o argumento nomeado e evitar TypeError
-                    await msg.update(content=f"❌ Erro na comunicação com o serviço de IA. Detalhe: {error_detail}")
+                    # CORREÇÃO AQUI TAMBÉM
+                    msg.content = f"❌ Erro na comunicação com o serviço de IA. Detalhe: {error_detail}"
+                    await msg.update()
                     return
 
-                # Itera sobre os pedaços (chunks) de texto recebidos do backend
+                # Streaming
                 async for chunk in response.aiter_text():
                     await msg.stream_token(chunk)
         
-        # Finaliza a mensagem. 
+        # Finaliza a mensagem (stream_token atualiza automaticamente, update final garante persistência)
         await msg.update()
 
     except httpx.ConnectError:
-        # Usa content= para forçar o argumento nomeado e evitar TypeError
-        await msg.update(content="Erro de conexão: Não foi possível se conectar ao backend. Verifique se ele está rodando.")
+        msg.content = "Erro de conexão: Não foi possível se conectar ao backend. Verifique se ele está rodando."
+        await msg.update()
     except Exception as e:
-        # Usa content= para forçar o argumento nomeado e evitar TypeError
-        await msg.update(content=f"Ocorreu um erro: {e}")
-        
-#--- END OF FILE app (25).py ---
+        msg.content = f"Ocorreu um erro: {e}"
+        await msg.update()
+
+#--- END OF FILE app.py ---