import gradio as gr
import openai, subprocess, os
from agents import agente
from dotenv import load_dotenv

openai.api_key = os.getenv('SECRET_API_KEY')

setupInicial, titulo = agente("po")

messages = [{"role": "system", "content": setupInicial}]

def chatea (pregunta, history):
	global messages
	
	messages.append({"role": "user", "content": pregunta})
	
	response = openai.chat.completions.create(model="gpt-4",messages=messages)
	
	system_message = response.choices[0].message.content
		
	return system_message
	
ui = gr.ChatInterface(fn=chatea, title=titulo, retry_btn=None, undo_btn=None, clear_btn=None, submit_btn="Enviar")
ui.launch(share=True)