Update app.py

app.py

@@ -1,17 +1,18 @@
 import os
-import chainlit as cl
-from groq import Groq
+import json
 import traceback
+import chainlit as cl
+import httpx  # Importamos a nova biblioteca
 
-# 1. Configuração do Cliente Groq
-client = Groq()
-
-# 2. Configurações do Modelo
+# 1. Configurações do Modelo e API
 MODEL_ID = "groq/compound"
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+API_URL = "https://api.groq.com/openai/v1/chat/completions"
 
-# 3. Início do Chat
+# 2. Início do Chat
 @cl.on_chat_start
 def start_chat():
+    """Define o prompt de sistema para guiar o comportamento do modelo."""
     system_prompt = (
         "Você é um assistente de pesquisa avançado. "
         "Quando um usuário perguntar sobre o conteúdo de um site (ex: 'o que tem no site X?'), "
@@ -24,9 +25,10 @@ def start_chat():
         [{"role": "system", "content": system_prompt}]
     )
 
-# 4. Lógica Principal de Mensagem (com Correção Final)
+# 3. Lógica Principal de Mensagem (usando httpx)
 @cl.on_message
 async def main(message: cl.Message):
+    """Processa a mensagem do usuário usando httpx para mais controle sobre o stream."""
     message_history = cl.user_session.get("message_history")
     message_history.append({"role": "user", "content": message.content})
 
@@ -35,37 +37,58 @@ async def main(message: cl.Message):
 
     full_response = ""
     try:
-        stream = client.chat.completions.create(
-            model=MODEL_ID,
-            messages=message_history,
-            temperature=0.7,
-            max_tokens=4096,
-            stream=True,
-            compound_custom={"tools": {"enabled_tools": ["web_search", "code_interpreter", "visit_website"]}}
-        )
+        # Montamos os headers e o payload manualmente
+        headers = {
+            "Authorization": f"Bearer {GROQ_API_KEY}",
+            "Content-Type": "application/json",
+        }
+        
+        payload = {
+            "model": MODEL_ID,
+            "messages": message_history,
+            "temperature": 0.7,
+            "max_tokens": 4096,
+            "stream": True,  # Essencial para o streaming
+            "compound_custom": {"tools": {"enabled_tools": ["web_search", "code_interpreter", "visit_website"]}}
+        }
 
-        for chunk in stream:
-            # O log de depuração pode ser comentado após confirmar que está tudo ok
-            # print(f"--- CHUNK BRUTO RECEBIDO ---\n{chunk.model_dump_json(indent=2)}\n----------------------------")
+        # Usamos um AsyncClient do httpx para fazer a requisição
+        async with httpx.AsyncClient(timeout=60) as client:
+            # client.stream nos dá um objeto de resposta para iterar
+            async with client.stream("POST", API_URL, headers=headers, json=payload) as response:
+                # Verificamos se a requisição foi bem-sucedida
+                response.raise_for_status()
+                
+                # Iteramos sobre as linhas da resposta (formato SSE)
+                async for line in response.aiter_lines():
+                    if line.startswith("data: "):
+                        # Removemos o prefixo "data: "
+                        json_data = line.removeprefix("data: ").strip()
+                        
+                        # O stream termina com um sinal "[DONE]"
+                        if json_data == "[DONE]":
+                            break
+                        
+                        # Convertemos o texto JSON em um dicionário Python
+                        chunk = json.loads(json_data)
 
-            if chunk.choices and chunk.choices[0].delta.content:
-                token = chunk.choices[0].delta.content
-                await msg.stream_token(token)
-                full_response += token
+                        # Extraímos o token de texto, se existir
+                        if chunk["choices"] and chunk["choices"][0]["delta"].get("content"):
+                            token = chunk["choices"][0]["delta"]["content"]
+                            await msg.stream_token(token)
+                            full_response += token
     
     except Exception as e:
-        print(f"--- ERRO DURANTE O STREAMING ---")
+        print(f"--- ERRO INESPERADO CAPTURADO ---")
         traceback.print_exc()
-        await msg.update(content=f"Desculpe, ocorreu um erro durante a comunicação com a API: {e}")
+        await msg.update(content=f"Desculpe, ocorreu um erro: {e}")
         return
 
-    # --- CORREÇÃO PRINCIPAL APLICADA AQUI ---
-    # Finalizamos a mensagem e o histórico de forma mais segura,
-    # garantindo que mesmo uma resposta vazia não quebre a sessão.
-    if full_response.strip():  # Verifica se a resposta não está vazia ou só com espaços
+    # Finaliza a mensagem e atualiza o histórico
+    if full_response.strip():
         await msg.update()
         message_history.append({"role": "assistant", "content": full_response})
         cl.user_session.set("message_history", message_history)
     else:
-        # Se nenhuma resposta foi gerada, removemos a mensagem vazia para limpar a interface.
-        await msg.remove()
+        # Se não houve resposta, informa o usuário e remove a mensagem vazia
+        await msg.update(content="O modelo não gerou uma resposta desta vez.")