better debug progress

app.py

@@ -13,9 +13,9 @@ import tempfile
 import ffmpeg
 import subprocess
 import traceback
-import time
 import shutil
 import LRC2SRT
+import sys
 
 flag = 1 
 
@@ -167,6 +167,23 @@ def stripinvisibles(s):
         e.replace(i,"")
     return e
 
+def start_progress(title):
+    global progress_x
+    sys.stdout.write(title + ": [" + "-"*40 + "]" + chr(8)*41)
+    sys.stdout.flush()
+    progress_x = 0
+
+def progress(x):
+    global progress_x
+    x = int(x * 40 // 100)
+    sys.stdout.write("#" * (x - progress_x))
+    sys.stdout.flush()
+    progress_x = x
+
+def end_progress():
+    sys.stdout.write("#" * (40 - progress_x) + "]\n")
+    sys.stdout.flush()
+
 haslyrics = False
 
 def main(file, name, fps=30, res: tuple=(1280,720), oscres=512, sr=11025, lyrics=None, img=None, tit=None, ast=None):
@@ -253,7 +270,7 @@ def main(file, name, fps=30, res: tuple=(1280,720), oscres=512, sr=11025, lyrics
     print('Frame Count: ' + str(num_frames))
     print('Segments per frame: ' + str(oscres))
     print('---------------------------------------------------------')
-    print('Rendering:')
+    start_progress("Rendering:")
     try:
         with Pool(cpu_count()) as pool:
             num_frames = len(samples_array) // (sr // fps)
@@ -261,10 +278,12 @@ def main(file, name, fps=30, res: tuple=(1280,720), oscres=512, sr=11025, lyrics
             for _ in pool.imap_unordered(render_frame, params):
                 iii += 1  # Increment frame count for progress
                 p((iii, num_frames), desc="Rendering Frames")
-                print(f'Frame {iii}/{num_frames}')
+                progress((iii/num_frames)*100)
                 
     except Exception as e:
         raise gr.Error("Something went wrong whilst rendering")
+    finally:
+        end_progress()
     
     p = gr.Progress()
     p(0.5, desc="Compiling video")