Update app.py

app.py

@@ -1,46 +1,59 @@
-import os
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Cliente com token seguro
-client = InferenceClient(model="meta-llama/Llama-3.3-70B-Instruct") 
-)
+# Cliente de inferência com modelo de IA pública
+client = InferenceClient(model="meta-llama/Llama-3.3-70B-Instruct")  # Modelo gratuito e avançado
 
-# Função de resposta no estilo chat
+# Função para processar a conversa
 def responder(mensagem, historico):
     mensagens = []
     if historico is None:
         historico = []
-    for u, b in historico:
-        mensagens.append({"role": "user", "content": u})
-        mensagens.append({"role": "assistant", "content": b})
+    
+    for item in historico:
+        if isinstance(item, list) and len(item) == 2:
+            user_msg, bot_msg = item
+            mensagens.append({"role": "user", "content": user_msg})
+            if bot_msg:
+                mensagens.append({"role": "assistant", "content": bot_msg})
+    
     mensagens.append({"role": "user", "content": mensagem})
-
     resposta = ""
 
     try:
-        for parte in client.chat_completion(messages=mensagens, stream=True):
-            # Debug: print parcial
-            print("[DEBUG] Parte recebida:", parte)
-            token = parte.get("choices", [{}])[0].get("delta", {}).get("content", "")
-            if token:
-                resposta += token
-                yield resposta
+        for mensagem in client.chat_completion(
+            mensagens,
+            max_tokens=300,
+            stream=True,
+            temperature=0.4,
+            top_p=0.8,
+        ):
+            if not mensagem or not isinstance(mensagem, dict):
+                continue
+            
+            try:
+                conteudo = mensagem["choices"][0]["delta"].get("content", "")
+                if conteudo.strip():
+                    resposta += conteudo
+                    yield resposta
+            except (AttributeError, IndexError, KeyError) as e:
+                print(f"Erro ao processar mensagem: {e}")
+                continue
+    
     except Exception as e:
-        print(f"[ERRO] {e}")
-        yield "Erro na geração da resposta."
+        print(f"Erro inesperado: {e}")
+        yield "Ocorreu um erro ao gerar a resposta."
 
     if not resposta.strip():
         yield "Nenhuma resposta gerada. Tente novamente."
 
-# Interface Gradio
+# Interface do chat com labels em português
 demo = gr.ChatInterface(
     responder,
-    title="Benjamin – Assistente Virtual da CEaD - IBC",
-    description="Tire dúvidas com minha inteligência artificial (minha base de dados vai até 2021)",
-    textbox=gr.Textbox(placeholder="Digite uma mensagem e tecle Enter"),
-    type="messages"
+    title="Benjamin – Assistente Virtual da CEaD - IBC. Tire dúvidas com minha inteligência artificial (minha base de dados vai até 2021)",
+    textbox=gr.Textbox(placeholder="Digite uma mensagem e depois tecle Enter"),
+    type="messages" 
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()