import openai
import os
import gradio as gr

openai.api_key = os.getenv("mykey2")

messages = [{"role": "system", "content": "You are a genius like Albert Einstein"}]


def CustomChatGPT(user_input):
    messages.append({"role": "user", "content": user_input})
    response = openai.ChatCompletion.create(
        model = "gpt-3.5-turbo",
        messages = messages
    )
    ChatGPT_reply = response["choices"][0]["message"]["content"]
    messages.append({"role": "assistant", "content": ChatGPT_reply})
    return ChatGPT_reply
    
description = "Cette Intelligence Artificielle est un programme fait par Stephane Dube (dubestephane @ vivaldi.net). Envoyez moi un message pour une requête ou pour une question. \n"
title = "Je peux répondre a presque toute les questions."

demo = gr.Interface(fn=CustomChatGPT, description = description, title = title, inputs = gr.inputs.Textbox(label = "Question", placeholder="Ex.: Quelle est la plus haute tour du monde?"), outputs = gr.outputs.Textbox(label = "Réponse"))
    
demo.launch()