Update app.py

app.py

@@ -11,9 +11,16 @@ users = os.getenv('LOGNAME')
 unames = users.split(',')
 pwds = os.getenv('PASSWORD')
 pwdList = pwds.split(',')
-dp = Path('/data')
-dp.mkdir(exist_ok=True)
-dataDir = '/data/'
+
+site = os.getenv('SITE')
+if site == 'local':
+    dp = Path('./data')
+    dp.mkdir(exist_ok=True)
+    dataDir = './data/'
+else:
+    dp = Path('/data')
+    dp.mkdir(exist_ok=True)
+    dataDir = '/data/'
 
 client = OpenAI(api_key = key)
 
@@ -40,10 +47,11 @@ def clear():
     return [None, [], None]
 
 def updatePassword(txt):
-    return [txt, "*********"]
+    return [txt.lower(), "*********"]
 
 def chat(prompt, user_window, pwd_window, past, response, gptModel):
-    if user_window == 'dave' and pwd_window == 'boss':
+    user_window = user_window.lower()
+    if user_window == unames[0] and pwd_window == pwdList[0]:
         if prompt == 'stats':
             response = genUsageStats()
             return [past, response, None]
@@ -78,7 +86,7 @@ with gr.Blocks() as demo:
     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.')
+    gr.Markdown('Enter user name & password then enter prompt and click submit button.  GPT 3.5 is cheaper but GPT 4o may perform better.  Restart conversation if topic changes')
     # heading = gr.Label(value="GPT Chat", scale=2, color="Crimson" )
     with gr.Row():
         user_window = gr.Textbox(label = "User Name")