Update app.py

app.py

@@ -1,59 +1,35 @@
-import gradio as gr
-from huggingface_hub import InferenceClient
-
-# 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
+from langchain.agents.openai_assistant import OpenAIAssistantRunnable
+import os
 
-# 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 = ""
+asst_id = os.getenv('assistant_key')
+#key = os.getenv('open_ai')
 
-    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."
+interpreter_assistant = OpenAIAssistantRunnable(assistant_id=asst_id)
+import gradio as gr
 
-    if not resposta.strip():
-        yield "Nenhuma resposta gerada. Tente novamente."
+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;
+}
+"""
 
-# 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" 
-)
+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")
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch(share=True)