io4it 0.0.0.9__tar.gz → 0.0.0.10__tar.gz

{io4it-0.0.0.9 → io4it-0.0.0.10}/PKG-INFO

@@ -1,8 +1,9 @@
 Metadata-Version: 2.1
 Name: io4it
-Version: 0.0.0.9
+Version: 0.0.0.10
 Home-page: 
 Author: 
 Author-email: 
 Keywords: orange3 add-on
+Provides-Extra: cuda
 License-File: License.txt

{io4it-0.0.0.9 → io4it-0.0.0.10}/io4it.egg-info/PKG-INFO

@@ -1,8 +1,9 @@
 Metadata-Version: 2.1
 Name: io4it
-Version: 0.0.0.9
+Version: 0.0.0.10
 Home-page: 
 Author: 
 Author-email: 
 Keywords: orange3 add-on
+Provides-Extra: cuda
 License-File: License.txt

{io4it-0.0.0.9 → io4it-0.0.0.10}/io4it.egg-info/requires.txt

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

{io4it-0.0.0.9 → io4it-0.0.0.10}/setup.py

@@ -1,10 +1,12 @@
 from setuptools import setup, find_packages
+import platform
+
 
 # Nom du package PyPI ('pip install NAME')
 NAME = "io4it"
 
 # Version du package PyPI
-VERSION = "0.0.0.9"  # la version doit être supérieure à la précédente sinon la publication sera refusée
+VERSION = "0.0.0.10"  # la version doit être supérieure à la précédente sinon la publication sera refusée
 
 # Facultatif / Adaptable à souhait
 AUTHOR = ""
@@ -33,6 +35,16 @@ PACKAGE_DATA = {
 # Dépendances
 INSTALL_REQUIRES = ["boto3", "docling", "docling-core", "speechbrain", "whisper", "whisper-openai", "pyannote.audio", "pyannote.core", "wave", "scikit-learn"]
 
+
+
+# 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": (
@@ -43,17 +55,29 @@ ENTRY_POINTS = {
 
 NAMESPACE_PACKAGES = ["orangecontrib"]
 
-setup(name=NAME,
-      version=VERSION,
-      author=AUTHOR,
-      author_email=AUTHOR_EMAIL,
-      url=URL,
-      description=DESCRIPTION,
-      license=LICENSE,
-      keywords=KEYWORDS,
-      packages=PACKAGES,
-      package_data=PACKAGE_DATA,
-      install_requires=INSTALL_REQUIRES,
-      entry_points=ENTRY_POINTS,
-      namespace_packages=NAMESPACE_PACKAGES,
-      )
+
+
+#   Message informatif si on est sous Windows
+if platform.system() == "Windows":
+    print("  Vous êtes sur Windows. Si vous avez une carte NVIDIA CUDA, vous pouvez installer les dépendances optimisées avec :")
+    print("   pip install io4it[cuda] --extra-index-url https://download.pytorch.org/whl/cu126")
+
+
+
+setup(
+    name=NAME,
+    version=VERSION,
+    author=AUTHOR,
+    author_email=AUTHOR_EMAIL,
+    url=URL,
+    description=DESCRIPTION,
+    license=LICENSE,
+    keywords=KEYWORDS,
+    packages=PACKAGES,
+    package_data=PACKAGE_DATA,
+    install_requires=INSTALL_REQUIRES,
+    extras_require=EXTRAS_REQUIRE,
+    entry_points=ENTRY_POINTS,
+    namespace_packages=NAMESPACE_PACKAGES,
+)
+