Update app.py

app.py

@@ -1,35 +1,59 @@
-from langchain.agents.openai_assistant import OpenAIAssistantRunnable
-import os
+import gradio as gr
+from huggingface_hub import InferenceClient
 
-asst_id = os.getenv('assistant_key')
-#key = os.getenv('open_ai')
+# 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
 
-interpreter_assistant = OpenAIAssistantRunnable(assistant_id=asst_id)
-import gradio as gr
+# Função para processar a conversa
+def responder(mensagem, historico):
+    mensagens = []
+    if historico is None:
+        historico = []
+    
+    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 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 inesperado: {e}")
+        yield "Ocorreu um erro ao gerar a resposta."
 
-def chat_response(message, history):
-  output = interpreter_assistant.invoke({"content": message})
-  response = output[0].content[0].text.value
-  return response
-css = """
-label[data-testid="block-label"] {
-    display: none !important;
-}
-footer {
-    display: none !important;
-}
-"""
+    if not resposta.strip():
+        yield "Nenhuma resposta gerada. Tente novamente."
 
-js_func = """
-function refresh() {
-    const url = new URL(window.location);
-    if (url.searchParams.get('__theme') !== 'dark') {
-        url.searchParams.set('__theme', 'dark');
-        window.location.href = url.href;
-    }
-}
-"""
-with gr.Blocks(css=css, js = js_func, theme="monochrome") as demo:
-  chatbot = gr.ChatInterface(chat_response,title="ProjSite")
+# Interface do chat com labels em português
+demo = gr.ChatInterface(
+    responder,
+    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" 
+)
 
-demo.launch(share=True)
+if __name__ == "__main__":
+    demo.launch()