doctra 0.3.1__py3-none-any.whl → 0.3.2__py3-none-any.whl

doctra/utils/progress.py

@@ -354,10 +354,11 @@ def create_notebook_friendly_bar(
     **kwargs
 ) -> tqdm:
     """
-    Create a notebook-friendly progress bar with minimal formatting.
+    Create a notebook-friendly progress bar with consistent sizing and static display.
     
     This function creates progress bars specifically optimized for Jupyter notebooks
-    to avoid display issues and ANSI code problems.
+    to avoid display issues and ANSI code problems while maintaining consistency
+    with the main progress bar styling.
     
     :param total: Total number of items to process
     :param desc: Description text for the progress bar
@@ -384,24 +385,52 @@ def create_notebook_friendly_bar(
         if prefix:
             desc = f"{prefix} {desc}"
     
-    # Simple format for notebooks
-    bar_format = "{desc}: {percentage:3.0f}%|{bar}| {n_fmt}/{total_fmt}"
+    # Use same format as main progress bar for consistency
+    bar_format = "{desc}: {percentage:3.0f}%|{bar}| {n_fmt}/{total_fmt} [{elapsed}<{remaining}]"
+    
+    # Color schemes based on operation type (same as main progress bar)
+    color_schemes = {
+        "loading": {"colour": "cyan", "ncols": 100},
+        "charts": {"colour": "green", "ncols": 100},
+        "tables": {"colour": "blue", "ncols": 100},
+        "figures": {"colour": "magenta", "ncols": 100},
+        "ocr": {"colour": "yellow", "ncols": 100},
+        "vlm": {"colour": "red", "ncols": 100},
+        "processing": {"colour": "white", "ncols": 100},
+    }
+    
+    # Determine color scheme based on description
+    if "loading" in desc_lower or "model" in desc_lower:
+        color_scheme = color_schemes["loading"]
+    elif "chart" in desc_lower:
+        color_scheme = color_schemes["charts"]
+    elif "table" in desc_lower:
+        color_scheme = color_schemes["tables"]
+    elif "figure" in desc_lower:
+        color_scheme = color_schemes["figures"]
+    elif "ocr" in desc_lower:
+        color_scheme = color_schemes["ocr"]
+    elif "vlm" in desc_lower:
+        color_scheme = color_schemes["vlm"]
+    else:
+        color_scheme = color_schemes["processing"]
     
     tqdm_config = {
         "total": total,
         "desc": desc,
         "leave": True,
         "bar_format": bar_format,
-        "ncols": _PROGRESS_CONFIG.ncols_env or 80,
+        "ncols": _PROGRESS_CONFIG.ncols_env or color_scheme["ncols"],  # Use same width as main progress bar
         "ascii": kwargs.get("ascii", False),
-        "dynamic_ncols": False,  # Fixed width for notebooks
-        "smoothing": 0.1,  # Faster updates
-        "mininterval": 0.05,
-        "maxinterval": 0.5,
+        "dynamic_ncols": True,  # Enable responsive width like main progress bar
+        "smoothing": 0.3,  # Use same smoothing as main progress bar
+        "mininterval": 0.1,  # Use same intervals as main progress bar
+        "maxinterval": 1.0,
         **kwargs
     }
     
-    return tqdm_auto(**tqdm_config)
+    # Use regular tqdm instead of tqdm_auto to avoid interactive widgets
+    return tqdm(**tqdm_config)
 
 
 def progress_for(iterable: Iterable[Any], desc: str, total: Optional[int] = None, leave: bool = True, **kwargs) -> Iterator[Any]:

doctra/version.py

@@ -1,2 +1,2 @@
 """Version information for Doctra."""
-__version__ = '0.3.1'
+__version__ = '0.3.2'

{doctra-0.3.1.dist-info → doctra-0.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: doctra
-Version: 0.3.1
+Version: 0.3.2
 Summary: Parse, extract, and analyze documents with ease
 Home-page: https://github.com/AdemBoukhris457/Doctra
 Author: Adem Boukhris

{doctra-0.3.1.dist-info → doctra-0.3.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 doctra/__init__.py,sha256=ST_c2GWBoB0y_wpL1qsOeK4bR1RyJhMMn6I5VjVRI6Y,613
-doctra/version.py,sha256=BDWZqR8pRPnlsqLDR4Kx91MC6A9OwylJHhHemdaa6DQ,60
+doctra/version.py,sha256=ioiNbDzSUf2zah2WhqG2TDJrYEKwp5tlGXIhVsPeCWE,62
 doctra/cli/__init__.py,sha256=4PTujjYRShOOUlZ7PwuWckShPWLC4v4CYIhJpzgyv1k,911
 doctra/cli/main.py,sha256=o_W1b5kx3xaTbWK6l4IYi0YLwffKBj5pQKflnlaG2Fw,35611
 doctra/cli/utils.py,sha256=IghiUZQCOmXODC5-5smHGz2KeV4xqbP4avmA1Mggln0,11800
@@ -34,11 +34,11 @@ doctra/utils/file_ops.py,sha256=3IS0EQncs6Kaj27fcg2zxQX3xRSvtItIsyKGLYgeOgw,815
 doctra/utils/io_utils.py,sha256=L1bWV4-ybs2j_3ZEN7GfQVgdC73JKVECVnpwKbP0dy0,219
 doctra/utils/ocr_utils.py,sha256=Doa1uYBg3kRgRYd2aPq9fICHgHfrM_efdhZfI7jl6OM,780
 doctra/utils/pdf_io.py,sha256=c8EY47Z1iqVtlLFHS_n0qGuXJ5ERFaMUd84ivXV0b9E,706
-doctra/utils/progress.py,sha256=sNEjTdN32J1-eXFPqwZRw2EZQ1SXSesXBd5StJvtlmc,14481
+doctra/utils/progress.py,sha256=Reo72IyKGTqUYu-956A2PD9hIVTDNrmmDbh4r_ie5Xo,15942
 doctra/utils/quiet.py,sha256=5XPS-1CtJ0sVk6qgSQctdhr_wR8mP1xoJLoUbmkXROA,387
 doctra/utils/structured_utils.py,sha256=J-qTqo8eCjm36FaRJ_I482LFgYCpm3eukZm-gbNnchw,1401
-doctra-0.3.1.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-doctra-0.3.1.dist-info/METADATA,sha256=2-2aMiNRvofe2WYuYejI6NqSkVctiH5SLK-EX4nIjaE,28298
-doctra-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-doctra-0.3.1.dist-info/top_level.txt,sha256=jI7E8jHci2gP9y0GYaWxlg9jG0O5n3FjHJJPLXDXMds,7
-doctra-0.3.1.dist-info/RECORD,,
+doctra-0.3.2.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+doctra-0.3.2.dist-info/METADATA,sha256=hXB4Lhcs9nuBrnHv1fPkaJz9C3Qr0BDiZfmCQTxfXys,28298
+doctra-0.3.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+doctra-0.3.2.dist-info/top_level.txt,sha256=jI7E8jHci2gP9y0GYaWxlg9jG0O5n3FjHJJPLXDXMds,7
+doctra-0.3.2.dist-info/RECORD,,