dlflannery commited on
Commit
a42a483
·
verified ·
1 Parent(s): 79615b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -27,6 +27,8 @@ client = OpenAI(api_key = key)
27
 
28
  def genUsageStats(do_reset=False):
29
  result = []
 
 
30
  for user in unames:
31
  tokens4o = 0
32
  tokens4mini = 0
@@ -43,10 +45,13 @@ def genUsageStats(do_reset=False):
43
  for line in dataList:
44
  (u, t) = line.split(':')
45
  (t, m) = t.split('-')
 
46
  if 'mini' in m:
47
- tokens4mini += int(t)
 
48
  else:
49
- tokens4o += int(t)
 
50
  accessOk = True
51
  break
52
  except:
@@ -54,6 +59,8 @@ def genUsageStats(do_reset=False):
54
  if not accessOk:
55
  return f'File access failed reading stats for user: {user}'
56
  result.append([user, str(tokens4mini), str(tokens4o)])
 
 
57
  return result
58
 
59
  def clear():
 
27
 
28
  def genUsageStats(do_reset=False):
29
  result = []
30
+ ttotal4o = 0
31
+ ttotal4mini = 0
32
  for user in unames:
33
  tokens4o = 0
34
  tokens4mini = 0
 
45
  for line in dataList:
46
  (u, t) = line.split(':')
47
  (t, m) = t.split('-')
48
+ tcount = int(t)
49
  if 'mini' in m:
50
+ tokens4mini += tcount
51
+ ttotal4mini += tcount
52
  else:
53
+ tokens4o += tcount
54
+ ttotal4o += tcount
55
  accessOk = True
56
  break
57
  except:
 
59
  if not accessOk:
60
  return f'File access failed reading stats for user: {user}'
61
  result.append([user, str(tokens4mini), str(tokens4o)])
62
+ gtotal = ttotal4mini + ttotal4o
63
+ result.append(['totals', str(ttotal4mini), str(ttotal4o), str(gtotal)])
64
  return result
65
 
66
  def clear():