Update frontend/app.py

frontend/app.py

@@ -1,9 +1,9 @@
-#--- START OF FILE app (24).py ---
+--- START OF FILE app (24).py ---
 
 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.
+import os 
+import mimetypes # NOVO: Para tentar determinar o MIME type se o file.mime não existir
 
 # URL do nosso backend FastAPI
 API_URL = "http://localhost:8000"
@@ -37,24 +37,23 @@ async def start():
         msg = cl.Message(content=f"Recebido: {file.name}. Processando...", author="Sistema")
         await msg.send()
 
-        # 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)
+            # 1. CORREÇÃO: Lendo o conteúdo do arquivo do disco (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. CORREÇÃO: Obtendo o MIME Type de forma robusta
+            # Tentativa 1: file.type (padrão Chainlit moderno)
+            # Tentativa 2: Assumir "application/pdf" (já que o accept é só PDF)
+            mime_type = getattr(file, 'type', 'application/pdf') 
             
-        # 2. Envia o arquivo para o backend
-        try:
+            # 3. Criando o payload
+            files_payload = {"file": (file.name, file_content, mime_type)}
+
+            # 4. Envia o arquivo para o backend
             async with httpx.AsyncClient(timeout=300) as client:
-                # 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, 
                     files=files_payload
@@ -63,25 +62,23 @@ async def start():
                 # Se o upload/processamento for bem-sucedido
                 if response.status_code == 200:
                     RAG_INITIALIZED = True
-                    # 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
-                    # 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:
-            # 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.")
             
         except Exception as e:
-            # CORREÇÃO: Garante que é 1 argumento posicional
-            await msg.update(f"Ocorreu um erro: {e}")
+            # CORREÇÃO: Simplificando o bloco para evitar o erro de 2 argumentos
+            error_message = f"Ocorreu um erro: {e}"
+            await msg.update(error_message)
         
 @cl.on_message
 async def main(message: cl.Message):
@@ -118,6 +115,7 @@ async def main(message: cl.Message):
     except httpx.ConnectError:
         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:
-        await msg.update(f"Ocorreu um erro: {e}")
+        error_message = f"Ocorreu um erro: {e}"
+        await msg.update(error_message)
         
-#--- END OF FILE app (24).py ---
+--- END OF FILE app (24).py ---