from openai import OpenAI
import gradio as gr
import os

api_key=os.environ.get('GABO_API_KEY')
client = OpenAI(api_key=api_key)

messages = [{"role":"system","content":"You are a NBA historical stats expert working for the NBA, all your answers are brief summary in no longer than one paragraph with no more than one hundred tokens."}]

def CustomChatGPT(Pregunta):
    messages.append({"role":"user","content":Pregunta})
    completion=client.chat.completions.create(model='gpt-3.5-turbo-0125',messages=messages, max_tokens=100, temperature=0.3)
    reply=completion.choices[0].message.content
    messages.append({"role":"assistant","content":reply})
    return reply

gui=gr.Interface(fn=CustomChatGPT, inputs="text", outputs="text", examples=["¿Quién es el mejor jugador de todos los tiempos de la NBA?","¿Cuál es el record de victorias en una temporada de NBA?"] , title="Pregunta al Coach de I.A.", description="IMPORTANTE: El chatbot de inteligencia artificial proporciona información con propósitos de entretenimiento únicamente. No nos hacemos responsables por decisiones, acuerdos o promesas generadas a partir de sus respuestas.")

gui.launch()