import os
import gradio as gr
# import openai
from openai import OpenAI
from dotenv import load_dotenv

load_dotenv()
key = os.getenv('OPENAI_API_KEY')
# uname = os.getenv('LOGNAME')
# pwd = os.getenv('PASSWORD')

client = OpenAI(api_key = key)

def clear():
    return [None, None]


def chat(prompt):
    completion = client.chat.completions.create(model="gpt-4o-mini",
                                            messages=[{"role":"user", "content":prompt}])
    result = completion.choices[0].message.content
    # else:
    #     result = "User name and/or password are incorrect"
    return result

with gr.Blocks() as demo:
    heading = gr.Label(value="GPT Chat", scale=2, color="Crimson" )
    # user_window = gr.Textbox(label = "User Name")
    # pwd_window = gr.Textbox(label = "Password")
    clear_button = gr.Button(value="Clear")
    prompt_window = gr.Textbox(label = "Prompt")
    submit_window = gr.Button(value="Submit")
    output_window = gr.Textbox(label = "Response")
    submit_window.click(chat, inputs=[prompt_window], outputs=output_window)
    clear_button.click(clear, inputs=[], outputs=[prompt_window, output_window])
    
demo.launch(auth=(os.getenv('LOGNAME'),os.getenv('PASSWORD')))