Update app.py

added perm files list.

app.py

@@ -3,7 +3,7 @@ from re import L
 import tempfile
 import gradio as gr
 # import openai
-from numpy._core.defchararray import isdecimal
+from numpy._core.defchararray import endswith, isdecimal
 from openai import OpenAI
 from dotenv import load_dotenv
 from pathlib import Path
@@ -147,6 +147,7 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel):
         isBoss = True
         if prompt == 'stats':
             response = genUsageStats()
+            list_permanent_files()
             return [past, response, None, gptModel]
         if prompt == 'reset':
             response = genUsageStats(True)
@@ -159,6 +160,10 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel):
             response = f'cleaned all .wav files for {user}'
             final_clean_up(user)
             return [past, response, None, gptModel]
+        if prompt.startswith('files'):
+            (log_cnt, wav_cnt, other_cnt, others) = list_permanent_files()
+            response = f'{log_cnt} log files\n{wav_cnt} .wav files\n{other_cnt} Other files:\n{others}'
+            return [past, response, None, gptModel]
     if user_window in unames and pwd_window == pwdList[unames.index(user_window)]:
         past.append({"role":"user", "content":prompt})
         completion = client.chat.completions.create(model=gptModel,
@@ -249,17 +254,38 @@ def gen_speech_file_names(user, cnt):
     return rv
 
 def final_clean_up(user):
-    flist = glob(dataDir + f'{user}_speech*.wav')
+    if user.strip().lower() == 'all':
+        flist = glob(dataDir + '*_speech*.wav')
+    else:
+        flist = glob(dataDir + f'{user}_speech*.wav')
     for fpath in flist:
         try:
             os.remove(fpath)
         except:
             continue
 
+
+def list_permanent_files():
+    flist = os.listdir(dataDir)
+    others = []
+    log_cnt = 0
+    wav_cnt = 0
+    other_cnt = 0
+    for fpath in flist:
+        if fpath.endswith('.txt'):
+            log_cnt += 1
+        elif fpath.endswith('.wav'):
+            wav_cnt += 1
+        else:
+            others.append(fpath)
+    other_cnt = len(others)
+    return (str(log_cnt), str(wav_cnt), str(other_cnt), str(others))
+
+
 with gr.Blocks() as demo:
     history = gr.State([])
     password = gr.State("")
-    user = gr.State("")
+    user = gr.State("unknown")
     model = gr.State("gpt-4o-mini")
     q = gr.State([])
     qsave = gr.State([])
@@ -350,10 +376,11 @@ with gr.Blocks() as demo:
             fp.write(response.content)
         return [fname, q]
 
-    def gen_output_audio(q):
+    def gen_output_audio(q, user):
         try:
             fname = q.pop(0)
         except:
+            final_clean_up(user)
             return [None, gr.Audio(sources=None)]
         return [fname, q]
 
@@ -388,8 +415,8 @@ with gr.Blocks() as demo:
     audio_widget.pause_recording(fn=pause_message, outputs=[prompt_window])
     reset_button.add(audio_widget)
     audio_out = gr.Audio(autoplay=True, visible=False)
-    audio_out.stop(fn=gen_output_audio, inputs=q, outputs = [audio_out, q])
+    audio_out.stop(fn=gen_output_audio, inputs=[q, user], outputs = [audio_out, q])
     speak_output.click(fn=initial_audio_output, inputs=[output_window, user_window], outputs=[audio_out, q])
     output_window.change(fn=set_speak_button, inputs=output_window,outputs=speak_output)
-    demo.unload(final_clean_up(user_window))
+    # demo.unload(final_clean_up(user))
 demo.launch(share=True)