io4it 0.0.0.11__tar.gz → 0.0.0.12.1__tar.gz

{io4it-0.0.0.11 → io4it-0.0.0.12.1}/PKG-INFO

@@ -1,6 +1,7 @@
 Metadata-Version: 2.1
 Name: io4it
-Version: 0.0.0.11
+Version: 0.0.0.12.1
+Summary: Package nécessitant PyTorch CUDA. Après installation, exécutez:
 Home-page: 
 Author: 
 Author-email: 

{io4it-0.0.0.11 → io4it-0.0.0.12.1}/io4it.egg-info/PKG-INFO

@@ -1,6 +1,7 @@
 Metadata-Version: 2.1
 Name: io4it
-Version: 0.0.0.11
+Version: 0.0.0.12.1
+Summary: Package nécessitant PyTorch CUDA. Après installation, exécutez:
 Home-page: 
 Author: 
 Author-email: 

{io4it-0.0.0.11 → io4it-0.0.0.12.1}/orangecontrib/IO4IT/ocr_function/word_converter.py

@@ -17,7 +17,7 @@ def enable_long_path(path):
     """Simplifie la gestion des chemins longs sous Windows."""
     return pathlib.Path(r"\\?\\" + str(path))
 
-def convert_pdf_structure(input_dir: str, output_dir: str,ignore_exsting_out_put=False,eprogress_callback=None):
+def convert_pdf_structure(input_dir: str, output_dir: str,ignore_exsting_out_put=False,progress_callback=None):
     """
     return a string with log in  case of error
     Recursively lists all .pdf and .PDF files in the input directory,

{io4it-0.0.0.11 → io4it-0.0.0.12.1}/setup.py

@@ -1,12 +1,35 @@
 from setuptools import setup, find_packages
 import platform
 import warnings
+# orangecontrib/IO4IT/__init__.py
+from setuptools import setup, find_packages
+import sys
+import subprocess
+from setuptools.command.install import install
+
+class PostInstallCommand(install):
+    """Post-installation pour l'installation GPU."""
+    def run(self):
+        install.run(self)  # Installation normale
+        
+        # Tentative d'installation PyTorch CUDA après l'installation principale
+        try:
+            subprocess.check_call([
+                sys.executable, '-m', 'pip', 'install',
+                'torch', 'torchvision', 'torchaudio',
+                '--index-url', 'https://download.pytorch.org/whl/cu126'
+            ])
+            print("\n\n✅ PyTorch CUDA installé avec succès")
+        except subprocess.CalledProcessError:
+            print("\n\n⚠️ IMPORTANT : Pour l'accélération GPU, exécutez MANUELLEMENT:")
+            print("pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126")
+
 
 # Nom du package PyPI ('pip install NAME')
 NAME = "io4it"
 
 # Version du package PyPI
-VERSION = "0.0.0.11"  # la version doit être supérieure à la précédente sinon la publication sera refusée
+VERSION = "0.0.0.12.1"  # la version doit être supérieure à la précédente sinon la publication sera refusée
 
 # Facultatif / Adaptable à souhait
 AUTHOR = ""
@@ -33,9 +56,20 @@ PACKAGE_DATA = {
 # /!\ les noms de fichier 'orangecontrib.hkh_bot.widgets' doivent correspondre à l'arborescence
 
 # Dépendances
-INSTALL_REQUIRES = ["boto3", "docling", "docling-core", "speechbrain", "whisper", "whisper-openai", "pyannote.audio", "pyannote.core", "wave", "scikit-learn"]
-
-
+# Dans votre setup.py principal
+INSTALL_REQUIRES = [
+    "boto3",
+    "docling",
+    "docling-core", 
+    "speechbrain",
+    "whisper",
+    "whisper-openai",
+    "pyannote.audio",
+    "pyannote.core",
+    "wave",
+    "scikit-learn",
+    # NE PAS inclure torch ici
+]
 
 # Extras optionnels pour CUDA
 EXTRAS_REQUIRE = {
@@ -72,7 +106,6 @@ setup(
     author=AUTHOR,
     author_email=AUTHOR_EMAIL,
     url=URL,
-    description=DESCRIPTION,
     license=LICENSE,
     keywords=KEYWORDS,
     packages=PACKAGES,
@@ -81,5 +114,7 @@ setup(
     extras_require=EXTRAS_REQUIRE,
     entry_points=ENTRY_POINTS,
     namespace_packages=NAMESPACE_PACKAGES,
+    description="Package nécessitant PyTorch CUDA. Après installation, exécutez:\n"
+               "pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126",
 )