Spaces:
Running
Running
File size: 2,956 Bytes
770f5f7 20d9410 caebb7e 770f5f7 20d9410 770f5f7 3de8d87 c9ffda4 1125ceb 770f5f7 20190f7 770f5f7 2a9567a 3b847a5 2a9567a a157f63 3de8d87 a157f63 3de8d87 3b847a5 a157f63 3b847a5 a157f63 3b847a5 067746e 35f5401 3de8d87 93675e1 ffcb5f5 770f5f7 2a9567a 3b847a5 a157f63 8819714 a157f63 8819714 3de8d87 a157f63 f6fe9f4 3b847a5 a157f63 3de8d87 3b847a5 2a9567a 20190f7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
import os
import gradio as gr
# import openai
from openai import OpenAI
from dotenv import load_dotenv
from pathlib import Path
load_dotenv()
key = os.getenv('OPENAI_API_KEY')
users = os.getenv('LOGNAME')
unames = users.split(',')
pwds = os.getenv('PASSWORD')
pwdList = pwds.split(',')
#cwd = os.getcwd()
client = OpenAI(api_key = key)
dataLogFilePath = "/home/user/data/usageData.txt"
def clear():
return [None, [], None]
def updatePassword(txt):
return [txt, "*********"]
def chat(prompt, user_window, pwd_window, past, response, gptModel):
if user_window in unames and pwd_window in pwdList:
past.append({"role":"user", "content":prompt})
completion = client.chat.completions.create(model=gptModel,
messages=past)
reply = completion.choices[0].message.content
response += "\n\nYOU: " + prompt + "\nGPT: " + reply
if len(response) > 40000:
response += "\n\nTHIS DIALOG IS GETTING TOO LONG. PLEASE CLEAR IT."
past.append({"role":"assistant", "content": reply})
try:
with open(dataLogFilePath, 'a') as f:
f.write(f'{user_window}: {len(response)} chars\n')
with open(dataLogFilePath) as f:
response += '\n' + f.read()
except Exception as e:
response += f"\nDATA LOG FAILED, path = {dataLogFilePath},\nmsg = {e}"
return [past, response, None]
else:
return [[], "User name and/or password are incorrect" + '\npwds = ' + ','.join(pwdList) + 'pwd= ' + pwd_window + ' user= '+user_window, prompt]
with gr.Blocks() as demo:
history = gr.State([])
password = gr.State("")
model = gr.State("gpt-3.5-turbo")
gr.Markdown('# GPT Chat')
gr.Markdown('Enter user name & password then enter prompt and click submit button. GPT 3.5 is cheaper but GPT 4o may perform better.')
# heading = gr.Label(value="GPT Chat", scale=2, color="Crimson" )
with gr.Row():
user_window = gr.Textbox(label = "User Name")
pwd_window = gr.Textbox(label = "Password")
pwd_window.blur(updatePassword, pwd_window, [password, pwd_window])
with gr.Row():
clear_button = gr.Button(value="Restart Conversation")
gpt_chooser=gr.Radio(choices=[("GPT-3.5","gpt-3.5-turbo"),("GPT-4o","gpt-4o-mini")],
value="gpt-3.5-turbo", label="GPT Model", interactive=True)
submit_window = gr.Button(value="Submit Prompt/Question")
prompt_window = gr.Textbox(label = "Prompt or Question")
output_window = gr.Textbox(label = "Dialog")
submit_window.click(chat, inputs=[prompt_window, user_window, password, history, output_window, model],
outputs=[history, output_window, prompt_window])
clear_button.click(clear, inputs=[], outputs=[prompt_window, history, output_window])
demo.launch(allowed_paths=["/home/user/data/usageData.txt"])
|