Update app.py

add audio duration stuff

app.py

@@ -5,6 +5,7 @@ from openai import OpenAI
 from dotenv import load_dotenv
 from pathlib import Path
 from time import sleep
+import audioread
 
 load_dotenv(override=True)
 key = os.getenv('OPENAI_API_KEY')
@@ -31,6 +32,7 @@ def genUsageStats(do_reset=False):
     ttotal4o_out = 0
     ttotal4mini_in = 0
     ttotal4mini_out = 0
+    totalAudio = 0
     for user in unames:
         tokens4o_in = 0
         tokens4o_out = 0
@@ -68,8 +70,29 @@ def genUsageStats(do_reset=False):
                     sleep(3)
             if not accessOk:
                 return f'File access failed reading stats for user: {user}'
-        result.append([user, f'{tokens4mini_in}/{tokens4mini_out}', f'{tokens4o_in}/{tokens4o_out}'])
-    result.append(['totals', f'{ttotal4mini_in}/{ttotal4mini_out}', f'{ttotal4o_in}/{ttotal4o_out}'])
+        userAudio = 0
+        fp = dataDir + user + '_audio.txt'
+        if os.path.exists(fp):
+            accessOk = False
+            for i in range(3):
+                try:
+                    with open(fp) as f:
+                        dataList = f.readlines()
+                    if do_reset:
+                        os.remove(fp)
+                    else:
+                        for line in dataList:
+                            (dud, len) = line.split(':')
+                            userAudio += int(len)
+                        totalAudio += int(userAudio)
+                    accessOk = True
+                    break
+                except:
+                    sleep(3)
+            if not accessOk:
+                return f'File access failed reading audio stats for user: {user}'
+        result.append([user, f'{tokens4mini_in}/{tokens4mini_out}', f'{tokens4o_in}/{tokens4o_out}', f'audio:{userAudio}'])
+    result.append(['totals', f'{ttotal4mini_in}/{ttotal4mini_out}', f'{ttotal4o_in}/{ttotal4o_out}', f'audio:{totalAudio}'])
     return result       
 
 def clear():
@@ -141,6 +164,11 @@ def transcribe(user, pwd, fpath):
     pwd = pwd.lower().strip()
     if not (user in unames and pwd in pwdList):
         return 'Bad credentials'
+    with audioread.audio_open(fpath) as audio:
+        duration = int(audio.duration)
+        if duration > 0:
+            with open(dataDir + user + '_audio.txt','a') as f:
+                f.write(f'audio:{str(duration)}\n')
     with open(fpath,'rb') as audio_file:
         transcript = client.audio.transcriptions.create(
             model='whisper-1', file = audio_file ,response_format = 'text' )