Update frontend/app.py

frontend/app.py

@@ -2,6 +2,8 @@
 
 import chainlit as cl
 import httpx
+import os # NOVO: Necessário para manipulação de arquivos
+import aiofiles # Opcional, mas melhor para leitura assíncrona. Vamos usar `open` para simplificar.
 
 # URL do nosso backend FastAPI
 API_URL = "http://localhost:8000"
@@ -35,12 +37,23 @@ async def start():
         msg = cl.Message(content=f"Recebido: {file.name}. Processando...", author="Sistema")
         await msg.send()
 
-        # Envia o arquivo para o backend
+        # 1. CORREÇÃO: Lendo o conteúdo do arquivo do disco (file.path)
+        file_content = None
+        try:
+            # Lendo o arquivo binário. O Chainlit já salvou o arquivo em um caminho temporário (file.path)
+            with open(file.path, "rb") as f:
+                file_content = f.read()
+
+        except Exception as e:
+            # CORREÇÃO: Garante que é 1 argumento posicional
+            await msg.update(f"❌ Erro ao ler o arquivo no frontend: {e}")
+            return
+            
+        # 2. Envia o arquivo para o backend
         try:
             async with httpx.AsyncClient(timeout=300) as client:
-                # Usamos form-data para enviar o arquivo.
-                # Assumimos que 'file.content' está correto para a sua versão do Chainlit
-                files_payload = {"file": (file.name, file.content, file.mime)}
+                # CORREÇÃO: O payload agora usa o `file_content` lido
+                files_payload = {"file": (file.name, file_content, file.mime)}
                 
                 response = await client.post(
                     UPLOAD_URL, 
@@ -50,29 +63,25 @@ async def start():
                 # Se o upload/processamento for bem-sucedido
                 if response.status_code == 200:
                     RAG_INITIALIZED = True
-                    # CORRIGIDO: Removido 'author='
+                    # CORREÇÃO: Garante que é 1 argumento posicional
                     await msg.update(
                         f"✅ **{file.name}** processado com sucesso! Agora você pode fazer perguntas sobre o conteúdo do documento."
                     )
                 else:
                     RAG_INITIALIZED = False
-                    # CORRIGIDO: Removido 'author='
+                    # CORREÇÃO: Garante que é 1 argumento posicional
                     error_detail = response.json().get('detail', 'Erro desconhecido')
                     await msg.update(
                         f"❌ Erro ao processar o arquivo. Código: {response.status_code}. Detalhe: {error_detail}"
                     )
 
         except httpx.ConnectError:
-            # CORRIGIDO: Removido 'author='
+            # CORREÇÃO: Garante que é 1 argumento posicional
             await msg.update("Erro de conexão: Não foi possível se conectar ao backend. Verifique se ele está rodando.")
-            # Para manter o autor 'Sistema', enviamos uma nova mensagem
-            await cl.Message(content="Erro de conexão: Não foi possível se conectar ao backend. Verifique se ele está rodando.", author="Sistema").send()
             
         except Exception as e:
-            # CORRIGIDO: Removido 'author='
+            # CORREÇÃO: Garante que é 1 argumento posicional
             await msg.update(f"Ocorreu um erro: {e}")
-            # Para manter o autor 'Sistema', enviamos uma nova mensagem
-            await cl.Message(content=f"Ocorreu um erro: {e}", author="Sistema").send()
         
 @cl.on_message
 async def main(message: cl.Message):
@@ -96,7 +105,6 @@ async def main(message: cl.Message):
             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:
-                    # CORRIGIDO: Uso correto do msg.update(content)
                     await msg.update(f"Erro: {response.text}")
                     return
 
@@ -108,10 +116,8 @@ async def main(message: cl.Message):
         await msg.update()
 
     except httpx.ConnectError:
-        # CORRIGIDO: Uso correto do msg.update(content)
         await msg.update("Erro de conexão: Não foi possível se conectar ao backend. Verifique se ele está rodando.")
     except Exception as e:
-        # CORRIGIDO: Uso correto do msg.update(content)
         await msg.update(f"Ocorreu um erro: {e}")
         
 #--- END OF FILE app (24).py ---