Update frontend/app.py

frontend/app.py

@@ -1,15 +1,84 @@
+#--- START OF FILE app (24).py ---
+
 import chainlit as cl
 import httpx
 
 # URL do nosso backend FastAPI
-API_URL = "http://localhost:8000/chat"
+API_URL = "http://localhost:8000"
+UPLOAD_URL = f"{API_URL}/upload-document"
+CHAT_URL = f"{API_URL}/chat"
+
+# Flag para saber se o RAG está inicializado
+RAG_INITIALIZED = False
 
 @cl.on_chat_start
-def start():
-    cl.user_session.set("history", [])
+async def start():
+    global RAG_INITIALIZED
+    RAG_INITIALIZED = False
+
+    # Mensagem de instrução inicial
+    await cl.Message(
+        content="Olá! Para começar, por favor, **faça o upload de um arquivo PDF** para eu usar como contexto. Após o upload, poderemos conversar sobre o conteúdo do documento.",
+        author="Sistema"
+    ).send()
+
+    # Pede o upload do arquivo
+    files = await cl.AskFileMessage(
+        content="Faça o upload do PDF:",
+        accept=["application/pdf"],
+        max_size_mb=20,
+        timeout=180,
+    ).send()
+
+    if files:
+        file = files[0]
+        msg = cl.Message(content=f"Recebido: {file.name}. Processando...", author="Sistema")
+        await msg.send()
+
+        # Envia o arquivo para o backend
+        try:
+            async with httpx.AsyncClient(timeout=300) as client:
+                # Usamos form-data para enviar o arquivo
+                response = await client.post(
+                    UPLOAD_URL, 
+                    files={"file": (file.name, file.content, file.mime)}
+                )
+                
+                # Se o upload/processamento for bem-sucedido
+                if response.status_code == 200:
+                    RAG_INITIALIZED = True
+                    # CORREÇÃO: Uso correto do msg.update(content)
+                    await msg.update(
+                        f"✅ **{file.name}** processado com sucesso! Agora você pode fazer perguntas sobre o conteúdo do documento.",
+                        author="Sistema"
+                    )
+                else:
+                    RAG_INITIALIZED = False
+                    # CORREÇÃO: Uso correto do msg.update(content)
+                    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}",
+                        author="Sistema"
+                    )
+
+        except httpx.ConnectError:
+            # CORREÇÃO: 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.", author="Sistema")
+        except Exception as e:
+            # CORREÇÃO: Uso correto do msg.update(content)
+            await msg.update(f"Ocorreu um erro: {e}", author="Sistema")
 
 @cl.on_message
 async def main(message: cl.Message):
+    global RAG_INITIALIZED
+    
+    if not RAG_INITIALIZED:
+        await cl.Message(
+            content="Por favor, aguarde o processamento do arquivo ou recarregue a página para um novo upload.",
+            author="Sistema"
+        ).send()
+        return
+
     # Cria uma mensagem de resposta vazia para preencher com o stream
     msg = cl.Message(content="")
     await msg.send()
@@ -18,10 +87,11 @@ async def main(message: cl.Message):
         # 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:
+            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:
-                    await msg.update(content=f"Erro: {response.text}")
+                    # CORREÇÃO: Uso correto do msg.update(content)
+                    await msg.update(f"Erro: {response.text}")
                     return
 
                 # Itera sobre os pedaços (chunks) de texto recebidos do backend
@@ -32,6 +102,10 @@ async def main(message: cl.Message):
         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.")
+        # CORREÇÃO: 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:
-        await msg.update(content=f"Ocorreu um erro: {e}")
+        # CORREÇÃO: Uso correto do msg.update(content)
+        await msg.update(f"Ocorreu um erro: {e}")
+        
+#--- END OF FILE app (24).py ---