Update app.py

Added HD image stats

app.py

@@ -101,6 +101,7 @@ def genUsageStats(do_reset=False):
     totalAudio = 0
     totalSpeech = 0
     totalImages = 0
+    totalHdImages = 0
     if do_reset:
         dudPath = dataDir + '_speech.txt'
         if os.path.exists(dudPath):
@@ -185,6 +186,7 @@ def genUsageStats(do_reset=False):
             if not accessOk:
                 return f'File access failed reading speech stats for user: {user}'
         user_images = 0
+        user_hd_images = 0
         fp = image_count_path(user)
         if os.path.exists(fp):
             accessOk = False
@@ -196,17 +198,21 @@ def genUsageStats(do_reset=False):
                         os.remove(fp)
                     else:
                         for line in dataList:
-                            cnt = line.strip()
-                            user_images += int(cnt)
-                        totalImages += int(user_images)
+                            x = line.strip()
+                            if x == 'hd':
+                                user_hd_images += 1
+                                totalHdImages += 1
+                            else:
+                                user_images += 1
+                                totalImages += 1
                     accessOk = True
                     break
                 except:
                     sleep(3)
             if not accessOk:
                 return f'File access failed reading image gen stats for user: {user}'
-        result.append([user, f'{tokens4mini_in}/{tokens4mini_out}', f'{tokens4o_in}/{tokens4o_out}', f'audio:{userAudio}',f'speech:{userSpeech}', f'images:{user_images}'])
-    result.append(['totals', f'{ttotal4mini_in}/{ttotal4mini_out}', f'{ttotal4o_in}/{ttotal4o_out}', f'audio:{totalAudio}',f'speech:{totalSpeech}', f'images:{totalImages}'])
+        result.append([user, f'{tokens4mini_in}/{tokens4mini_out}', f'{tokens4o_in}/{tokens4o_out}', f'audio:{userAudio}',f'speech:{userSpeech}', f'images:{user_images}/{user_hd_images}'])
+    result.append(['totals', f'{ttotal4mini_in}/{ttotal4mini_out}', f'{ttotal4o_in}/{ttotal4o_out}', f'audio:{totalAudio}',f'speech:{totalSpeech}', f'images:{totalImages}/{totalHdImages}'])
     return result       
 
 def new_conversation(user):
@@ -452,7 +458,10 @@ def make_image(prompt, user, pwd):
                 fpath = dataDir + user + '.png'
                 image.save(fpath)
             with open(image_count_path(user), 'at') as fp:
-                fp.write('1\n')
+                if quality == 'hd':
+                    fp.write('hd')
+                else:
+                    fp.write('1\n')
             msg = 'Image created!'
         except:
             return [gr.Image(visible=False, value=None), msg]