Update app.py

app.py

@@ -27,7 +27,7 @@ client = OpenAI(api_key = key)
 def genUsageStats(do_reset=False):
     result = []
     for user in unames:
-        total = 0
+        tokens = 0
         fp = dataDir + user + '_log.txt'
         if os.path.exists(fp):
             with open(fp) as f:
@@ -35,12 +35,11 @@ def genUsageStats(do_reset=False):
             if do_reset:
                 os.remove(fp)
             for line in dataList:
-                (u, c) = line.split(':')
-                total += int(c)
+                (u, t) = line.split(':')
+                tokens += int(t)
         else:
             total = 0
-        result.append([user, str(total)])
-            
+        result.append([user, str(tokens)])
     return result       
 
 def clear():
@@ -54,7 +53,9 @@ def setModel(val):
 
 def chat(prompt, user_window, pwd_window, past, response, gptModel):
     user_window = user_window.lower()
+    isBoss = False
     if user_window == unames[0] and pwd_window == pwdList[0]:
+        isBoss = True
         if prompt == 'stats':
             response = genUsageStats()
             return [past, response, None]
@@ -66,14 +67,17 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel):
         completion = client.chat.completions.create(model=gptModel,
                                             messages=past)
         reply = completion.choices[0].message.content
-        response += "\n\nYOU: " + prompt + "\nGPT: " + reply # + "\nModel = " + gptModel
-        if len(response) > 40000:
-            response += "\n\nTHIS DIALOG IS GETTING TOO LONG. PLEASE CLEAR IT."
+        tokens = completion.usage.total_tokens
+        response += "\n\nYOU: " + prompt + "\nGPT: " + reply
+        if isBoss:
+            response += f"\n{gptModel}: {tokens} tokens"
+        if tokens > 40000:
+            response += "\n\nTHIS DIALOG IS GETTING TOO LONG. PLEASE RESTART CONVERSATION SOON."
         past.append({"role":"assistant", "content": reply})
         try:
             dataFile = dataDir + user_window + '_log.txt'
             with open(dataFile, 'a') as f:
-                f.write(f'{user_window}: {len(response)}\n')
+                f.write(f'{user_window}: {tokens}\n')
 #            with open(dataFile) as f:
 #                response += '\n' + f.read()
         except Exception as e: