Update app.py

app.py

@@ -28,7 +28,8 @@ client = OpenAI(api_key = key)
 def genUsageStats(do_reset=False):
     result = []
     for user in unames:
-        tokens = 0
+        tokens3 = 0
+        tokens4 = 0
         fp = dataDir + user + '_log.txt'
         if os.path.exists(fp):
             accessOk = False
@@ -40,7 +41,11 @@ def genUsageStats(do_reset=False):
                         os.remove(fp)
                     for line in dataList:
                         (u, t) = line.split(':')
-                        tokens += int(t)
+                        (t, m) = t.split('-')
+                        if '4' in m:
+                            tokens4 += int(t)
+                        else:
+                            tokens3 += int(t)
                     accessOk = True
                     break
                 except:
@@ -49,7 +54,7 @@ def genUsageStats(do_reset=False):
                 return f'File access failed reading stats for user: {user}'
         else:
             total = 0
-        result.append([user, str(tokens)])
+        result.append([user, str(tokens3), str(tokens4)])
     return result       
 
 def clear():
@@ -89,7 +94,10 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel):
             try:
                 dataFile = new_func(user_window)
                 with open(dataFile, 'a') as f:
-                    f.write(f'{user_window}: {tokens}\n')
+                    m = '3'
+                    if '4' in gptModel:
+                        m = '4'
+                    f.write(f'{user_window}: {tokens}-{m}\n')
                 accessOk = True
                 break
             except Exception as e: