added per-chunk progress bars

radis · minouHub · May 4, 2026 · Nov 14, 2025 · Nov 20, 2025 · Dec 7, 2025
commit db74bb681f2c0eb71cf32063064cbe6efe8aad7e
diff --git a/radis/api/hitempapi.py b/radis/api/hitempapi.py
@@ -612,6 +612,40 @@ def parse_one_CO2_block(
     return df
 
 
+def _download_single_chunk(
+    start_wavno, end_wavno, out_file, session, engine, verbose=True, position=0
+):
+    """Download and process a single CO2 chunk."""
+    try:
+        if engine == "vaex":
+            fcache = _fcache_file_name(out_file, engine)
+            fcache_str = str(fcache)
+            if os.path.exists(fcache_str.replace(".hdf5", ".h5")):
+                update_pytables_to_vaex(fcache_str.replace(".hdf5", ".h5"))
+
+        desc = f"Chunk {start_wavno:.0f}-{end_wavno:.0f} cm⁻¹"
+        with tqdm(
+            total=1, desc=desc, position=position, leave=True, disable=not verbose
+        ) as pbar:
+            partial_download_co2_chunk(
+                start_wavno,
+                end_wavno,
+                session,
+                out_file,
+                verbose=False,  # Suppress verbose to avoid cluttered output
+            )
+            pbar.update(1)
+        if verbose:
+            print(f"Completed chunk {start_wavno:.0f}-{end_wavno:.0f} cm⁻¹")
+        return True
+    except Exception as e:
+        if verbose:
+            print(
+                f" Error downloading chunk {start_wavno:.0f}-{end_wavno:.0f} cm⁻¹: {e}"
+            )
+        raise
+
+
 def read_and_write_chunked_for_CO2(
     load_wavenum_max,
     load_wavenum_min,
@@ -696,37 +730,6 @@ def _append_dataframe(df_to_append):
                 f"- Download {len(files_to_download)} file(s) missing out of {len(wav_pairs)}."
             )
 
-        def _download_single_chunk(start_wavno, end_wavno, out_file):
-            """Download and process a single CO2 chunk."""
-            try:
-                if engine == "vaex":
-                    fcache = _fcache_file_name(out_file, engine)
-                    fcache_str = str(fcache)
-                    if os.path.exists(fcache_str.replace(".hdf5", ".h5")):
-                        update_pytables_to_vaex(fcache_str.replace(".hdf5", ".h5"))
-
-                if verbose:
-                    print(f"→ Downloading chunk {start_wavno:.0f}-{end_wavno:.0f} cm⁻¹")
-
-                partial_download_co2_chunk(
-                    start_wavno,
-                    end_wavno,
-                    session,
-                    out_file,
-                    verbose=False,  # Suppress verbose to avoid cluttered output
-                )
-
-                if verbose:
-                    print(f" Completed chunk {start_wavno:.0f}-{end_wavno:.0f} cm⁻¹")
-                return True
-
-            except Exception as e:
-                if verbose:
-                    print(
-                        f" Error downloading chunk {start_wavno:.0f}-{end_wavno:.0f} cm⁻¹: {e}"
-                    )
-                raise
-
         # Use parallel downloads for multiple chunks
         if len(files_to_download) > 1:
             if verbose:
@@ -736,16 +739,26 @@ def _download_single_chunk(start_wavno, end_wavno, out_file):
 
             with ThreadPoolExecutor(max_workers=max_threads) as executor:
                 futures = [
-                    executor.submit(_download_single_chunk, start, end, out_file)
-                    for (start, end, out_file) in files_to_download
+                    executor.submit(
+                        _download_single_chunk,
+                        start,
+                        end,
+                        out_file,
+                        session,
+                        engine,
+                        verbose,
+                        idx,
+                    )
+                    for idx, (start, end, out_file) in enumerate(files_to_download)
                 ]
-
                 for future in as_completed(futures):
                     future.result()
         else:
             # Single chunk download - no threading overhead
             start_wavno, end_wavno, out_file = files_to_download[0]
-            _download_single_chunk(start_wavno, end_wavno, out_file)
+            _download_single_chunk(
+                start_wavno, end_wavno, out_file, session, engine, verbose, 0
+            )
 
     else:
         if verbose: