GradioTest / app.py
dlflannery's picture
Update app.py
2a9567a verified
raw
history blame
1.19 kB
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(username, password, prompt):
if username==uname and pwd==password:
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:
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=[user_window, pwd_window, prompt_window], outputs=output_window)
clear_button.click(clear, inputs=[], outputs=[prompt_window, output_window])
demo.launch()