improve progress bars

diff --git a/main.py b/main.py
index 915eea6cab2cbbb4d2a47151ea9538e47845fd20..bb3a40985c53c8b07e3937aa19c57c19aaa674e2 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -134,7 +134,7 @@ def do_work(file_processor, message):
                     if lib.tqdm_avail:
                         # A nice progress bar but read/write instead of copyfileobj slows it down a little.
                         total_size = int(r.headers.get('content-length', 0))
-                        progress_bar = lib.tqdm(desc="download", total=total_size, unit='iB', unit_scale=True)
+                        progress_bar = lib.tqdm(desc="download", total=total_size, unit='iB', unit_scale=True, leave=False)
                         with open(input_path, 'wb') as f:
                             for data in r.iter_content(100000):
                                 f.write(data)

diff --git a/sm.py b/sm.py
index b221c42606905beca09be8e518b2581d8f326290..fbd5505754ac2de904393ff6e8ecd7cf88fd7cc0 100644 (file)
--- a/sm.py
+++ b/sm.py
@@ -47,7 +47,7 @@ def extract_text(input_path, output_path, debug=False):
         # Extract the text from each (single-threaded)
         # INPUT.pdf -> INPUT.pdf.txt
         with lib.timer("pdftotext", logging.warning):
-            for pdf in lib.tqdm(pdfs, desc="pdftotext"):
+            for pdf in lib.tqdm(pdfs, desc="pdftotext", leave=False):
                 input_pdf = os.path.join(td_in, pdf)
                 output_txt = os.path.join(td_out, pdf + ".txt")
                 #logging.info("converting {} -> {}".format(input_pdf, output_txt))