Spaces:
Running
Running
Update app.py
Browse files
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 +=
|
|
|
48 |
else:
|
49 |
-
tokens4o +=
|
|
|
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():
|