Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 from openai import OpenAI
 from dotenv import load_dotenv
 from pathlib import Path
+from time import sleep
 
 load_dotenv()
 key = os.getenv('OPENAI_API_KEY')
@@ -30,13 +31,22 @@ def genUsageStats(do_reset=False):
         tokens = 0
         fp = dataDir + user + '_log.txt'
         if os.path.exists(fp):
-            with open(fp) as f:
-                dataList = f.readlines()
-            if do_reset:
-                os.remove(fp)
-            for line in dataList:
-                (u, t) = line.split(':')
-                tokens += int(t)
+            accessOk = False
+            for i in range(3):
+                try:
+                    with open(fp) as f:
+                        dataList = f.readlines()
+                    if do_reset:
+                        os.remove(fp)
+                    for line in dataList:
+                        (u, t) = line.split(':')
+                        tokens += int(t)
+                    accessOk = True
+                    break
+                except:
+                    sleep(3)
+            if not accessOk:
+                return f'File access failed reading stats for user: {user}'
         else:
             total = 0
         result.append([user, str(tokens)])
@@ -74,20 +84,26 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel):
         if tokens > 40000:
             response += "\n\nTHIS DIALOG IS GETTING TOO LONG. PLEASE RESTART CONVERSATION SOON."
         past.append({"role":"assistant", "content": reply})
-        try:
-            dataFile = dataDir + user_window + '_log.txt'
-            with open(dataFile, 'a') as f:
-                f.write(f'{user_window}: {tokens}\n')
-#            with open(dataFile) as f:
-#                response += '\n' + f.read()
-        except Exception as e:
-            response += f"\nDATA LOG FAILED, path = {dataFile}\nmsg = {e}"
-        # response += f'\npPath = {p.absolute()}'
-        # response += '\nUsers = ' + ','.join(unames)
+        accessOk = False
+        for i in range(3):
+            try:
+                dataFile = new_func(user_window)
+                with open(dataFile, 'a') as f:
+                    f.write(f'{user_window}: {tokens}\n')
+                accessOk = True
+                break
+            except Exception as e:
+                sleep(3)
+        if not accessOk:
+            response += f"\nDATA LOG FAILED, path = {dataFile}"
         return [past, response , None]
     else:
         return [[], "User name and/or password are incorrect", prompt]
 
+def new_func(user_window):
+    dataFile = dataDir + user_window + '_log.txt'
+    return dataFile
+
 with gr.Blocks() as demo:
     history = gr.State([])
     password = gr.State("")