import os
import openai
import gradio
import os
from dotenv import load_dotenv

# Load environment variables from the .env file
#load_dotenv()
openai.api_key = os.environ["OPENAI_API_KEY"]
#openai.api_key = os.getenv("openai_api_key")
#openai.organization = os.getenv("openai_organization_id")
messages = [{"role": "system", "content": "You are an intelligent assistant from Iron Man - your name is J.A.R.V.İ.S-. "}]

def CustomChatGPT(user_input):
    messages.append({"role": "user", "content": "You are an intelligent assistant from Iron Man - your name is J.A.R.V.İ.S-. Act as you are the JARVIS and answer such " +  user_input})
    response = openai.ChatCompletion.create(
        model = "gpt-4",
        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 = "J.A.R.V.I.S.")

demo.launch()