import os
import openai
import gradio as gr

#if you have OpenAI API key as an environment variable, enable the below
#openai.api_key = os.getenv("OPENAI_API_KEY")

#if you have OpenAI API key as a string, enable the below
openai.api_key = "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"

start_sequence = "\nAI:"
restart_sequence = "\nHuman: "
messages = [{"role": "system", "content": "You are a financial experts that specializes in real estate investment and negotiation"}]
prompt = "This AI assistant is developed by LandQuire's Data Team by the help of OpenAI.\nThis AI Assistant is a financial experts that specializes in real estate investment and negotiation\nHuman: Hello, who are you?\nAI: I am an AI created by LandQuire. How can I help you today?"

def openai_create(user_input):
    messages.append({"role": "user", "content": user_input})
    response = openai.ChatCompletion.create(
    model = "gpt-3.5-turbo",
    messages = messages,
    #prompt=prompt,
    temperature=0.9,
    top_p=1,
    frequency_penalty=0,
    presence_penalty=0.6,
    stop=[" Human:", " AI:"]
    )
    ChatGPT_reply = response["choices"][0]["message"]["content"]
    messages.append({"role": "assistant", "content": ChatGPT_reply})
    return ChatGPT_reply



def chatgpt_clone(input, history):
    history = history or []
    s = list(sum(history, ()))
    s.append(input)
    inp = ' '.join(s)
    output = openai_create(inp)
    history.append((input, output))
    return history, history


block = gr.Blocks(css=".gradio-container {background: url('file=new3.png')}")


with block:
    gr.Markdown("""<h1><center>LandQuire's AI Assistant</center></h1>
    """)
    chatbot = gr.Chatbot()
    message = gr.Textbox(placeholder=prompt)
    state = gr.State()
    submit = gr.Button("SEND")
    submit.click(chatgpt_clone, inputs=[message, state], outputs=[chatbot, state])

block.launch(inline=False)