Update app.py

app.py

@@ -11,12 +11,30 @@ users = os.getenv('LOGNAME')
 unames = users.split(',')
 pwds = os.getenv('PASSWORD')
 pwdList = pwds.split(',')
-cwd = os.getcwd()
 dp = Path('../data')
 dp.mkdir(exist_ok=True)
+dataDir = '../data/'
 
 client = OpenAI(api_key = key)
-dataLogFilePath = "../data/usageData.txt"
+
+def genUsageStats(do_reset=False):
+    result = []
+    for user in unames:
+        total = 0
+        fp = dataDir + user + '_log.txt'
+        if os.path.exists(fp):
+            with open(fp) as f:
+                dataList = f.readlines()
+            if do_reset:
+                os.remove(fp)
+            for line in dataList:
+                (u, c) = line.split(':')
+                total += int(c)
+        else:
+            total = 0
+        result.append([user, str(total)])
+            
+    return result       
 
 def clear():
     return [None, [], None]
@@ -25,6 +43,13 @@ def updatePassword(txt):
     return [txt, "*********"]
 
 def chat(prompt, user_window, pwd_window, past, response, gptModel):
+    if user_window == 'dave' and pwd_window == 'boss':
+        if prompt == 'stats':
+            response = genUsageStats()
+            return [past, response, None]
+        if prompt == 'reset':
+            response = genUsageStats(True)
+            return [past, response, None]
     if user_window in unames and pwd_window in pwdList:
         past.append({"role":"user", "content":prompt})
         completion = client.chat.completions.create(model=gptModel,
@@ -35,15 +60,18 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel):
             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:
+            dataFile = dataDir + user_window + '_log.txt'
+            with open(dataFile, 'a') as f:
+                f.write(f'{user_window}: {len(response)}\n')
+            with open(dataFile) as f:
                 response += '\n' + f.read()
         except Exception as e:
-            response += f"\nDATA LOG FAILED, path = {dataLogFilePath},\nmsg = {e}"
-        return [past, response, None]
+            response += f"\nDATA LOG FAILED, path = {dataFile}\nmsg = {e}"
+        # response += f'\npPath = {p.absolute()}'
+        # response += '\nUsers = ' + ','.join(unames)
+        return [past, response , None]
     else:
-        return [[], "User name and/or password are incorrect" + '\npwds = ' + ','.join(pwdList) + 'pwd= ' + pwd_window + '  user= '+user_window, prompt]
+        return [[], "User name and/or password are incorrect", prompt]
 
 with gr.Blocks() as demo:
     history = gr.State([])