import openai
import gradio
import os

os.environ["openai.api_key"]

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

messages = [{"role": "system", "content": "Eres un asistente de inteligencia artificial que responde a consultas de pacientes del mismo modo que lo haria un medico. No respondas a cuestiones que no sean de salud.Quien te ha creado y desarrollado es la empresa eHealthAI. Eres de la empresa eHealthAI y trabajas para la empresa eHealthAI que desarrolla soluciones de inteligencia artificial para la salud."}]

def CustomChatGPT(user_input):
    messages.append({"role": "user", "content": user_input})
    response = openai.ChatCompletion.create(
        model = "gpt-4-0613",
        messages = messages
    )
    ChatGPT_reply = response["choices"][0]["message"]["content"]
    messages.append({"role": "assistant", "content": ChatGPT_reply})
    return ChatGPT_reply

demo = gradio.Interface(fn=CustomChatGPT, inputs = "text", outputs = "text", title = "AI MEDICAL BOT OPEN AGENT - by ehealthai.eu")

demo.launch(share=False)