io4it 0.0.0.12.3__tar.gz → 0.0.0.12.4__tar.gz

io4it-0.0.0.12.4/PKG-INFO

@@ -0,0 +1,8 @@
+Metadata-Version: 2.1
+Name: io4it
+Version: 0.0.0.12.4
+Home-page: 
+Author: 
+Author-email: 
+Keywords: orange3 add-on
+License-File: License.txt

io4it-0.0.0.12.4/io4it.egg-info/PKG-INFO

@@ -0,0 +1,8 @@
+Metadata-Version: 2.1
+Name: io4it
+Version: 0.0.0.12.4
+Home-page: 
+Author: 
+Author-email: 
+Keywords: orange3 add-on
+License-File: License.txt

{io4it-0.0.0.12.3 → io4it-0.0.0.12.4}/io4it.egg-info/requires.txt

@@ -8,8 +8,3 @@ pyannote.audio
 pyannote.core
 wave
 scikit-learn
-
-[cuda]
-torch
-torchvision
-torchaudio

{io4it-0.0.0.12.3 → io4it-0.0.0.12.4}/setup.py

@@ -7,30 +7,63 @@ import sys
 import subprocess
 from setuptools.command.install import install
 
-class PostInstallCommand(install):
-    """Post-installation pour l'installation GPU."""
+from setuptools import setup, find_packages
+import sys
+import subprocess
+import platform
+from setuptools.command.install import install
+
+class CustomInstall(install):
     def run(self):
-        install.run(self)  # Installation normale
+        # Installation normale d'abord
+        install.run(self)
         
-        # Tentative d'installation PyTorch CUDA après l'installation principale
+        # Vérification CUDA seulement sur Windows
+        if platform.system() == "Windows":
+            self._install_torch_cuda()
+
+    def _install_torch_cuda(self):
+        try:
+            import torch
+            if torch.cuda.is_available():
+                print("\n PyTorch CUDA est déjà installé et fonctionnel")
+                return
+        except ImportError:
+            pass
+
+        print("\n🔧 Tentative d'installation de PyTorch CUDA 12.6...")
         try:
             subprocess.check_call([
                 sys.executable, '-m', 'pip', 'install',
-                'torch', 'torchvision', 'torchaudio',
-                '--index-url', 'https://download.pytorch.org/whl/cu126'
+                'torch==2.2.2+cu126',
+                'torchvision==0.17.2+cu126', 
+                'torchaudio==2.2.2+cu126',
+                '--index-url', 'https://download.pytorch.org/whl/cu126',
+                '--no-deps'  # Évite de réinstaller les dépendances existantes
             ])
             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")
-
+            print("\n\n ÉCHEC de l'installation automatique de PyTorch CUDA")
+            print("Veuillez exécuter MANUELLEMENT cette commande :")
+            print("pip install torch==2.2.2+cu126 torchvision==0.17.2+cu126 torchaudio==2.2.2+cu126 --index-url https://download.pytorch.org/whl/cu126")
 
-# Nom du package PyPI ('pip install NAME')
+# Configuration
 NAME = "io4it"
+VERSION = "0.0.0.12.4"  # Incrémentez la version
 
-# Version du package PyPI
-VERSION = "0.0.0.12.3"  # la version doit être supérieure à la précédente sinon la publication sera refusée
-
+INSTALL_REQUIRES = [
+    "boto3",
+    "docling",
+    "docling-core", 
+    "speechbrain",
+    "whisper",
+    "whisper-openai",
+    "pyannote.audio",
+    "pyannote.core",
+    "wave",
+    "scikit-learn",
+    # Exclure torch intentionnellement
+]
 # Facultatif / Adaptable à souhait
 AUTHOR = ""
 AUTHOR_EMAIL = ""
@@ -71,14 +104,6 @@ INSTALL_REQUIRES = [
     # NE PAS inclure torch ici
 ]
 
-# Extras optionnels pour CUDA
-EXTRAS_REQUIRE = {
-    "cuda": [
-        "torch",
-        "torchvision",
-        "torchaudio"
-    ]
-}
 # Spécifie le dossier contenant les widgets et le nom de section qu'aura l'addon sur Orange
 ENTRY_POINTS = {
     "orange.widgets": (
@@ -111,14 +136,10 @@ setup(
     packages=PACKAGES,
     package_data=PACKAGE_DATA,
     install_requires=INSTALL_REQUIRES,
-    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",
-    
     cmdclass={
-        'install': PostInstallCommand,
+        'install': CustomInstall,
     },
 )
 

io4it-0.0.0.12.3/PKG-INFO

@@ -1,10 +0,0 @@
-Metadata-Version: 2.1
-Name: io4it
-Version: 0.0.0.12.3
-Summary: Package nécessitant PyTorch CUDA. Après installation, exécutez:
-Home-page: 
-Author: 
-Author-email: 
-Keywords: orange3 add-on
-Provides-Extra: cuda
-License-File: License.txt

io4it-0.0.0.12.3/io4it.egg-info/PKG-INFO

@@ -1,10 +0,0 @@
-Metadata-Version: 2.1
-Name: io4it
-Version: 0.0.0.12.3
-Summary: Package nécessitant PyTorch CUDA. Après installation, exécutez:
-Home-page: 
-Author: 
-Author-email: 
-Keywords: orange3 add-on
-Provides-Extra: cuda
-License-File: License.txt