Merge pull request #3 from UMass-Rescue/rescubox-flaskml-conformant

Rescuebox flaskml conformant

Author: sahilx13

.gitignore

@@ -160,4 +160,11 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
 
-.ruff_cache/
+.ruff_cache/
+
+# project specific file ignores
+rescuebox/bin/
+rescuebox/lib/
+rescuebox/pyvenv.cfg
+src/rb-api/rb/api/static/index/main.js
+**/*/.DS_Store

poetry.lock

@@ -13,13 +13,18 @@ requests = "^2.32.3"
 python = ">=3.11,<3.13"
 pyyaml = "^6.0.2"
 typer = "^0.12.5"
+llvmlite = "^0.44.0"
+numba = "^0.61.0"
 
 rb-lib = { path = "src/rb-lib", develop = true }
 
 rb-file-utils = { path = "src/rb-file-utils", develop = true }
 rb-doc-parser = { path = "src/rb-doc-parser", develop = true }
+rb-audio-transcription = { path = "src/rb-audio-transcription", develop = true }
 
 # Don't add new packages here, add them appropriately in the list above
+torch = "2.2.2"
+numpy = "1.26.4"
 
 
 [tool.poetry.group.dev.dependencies]
@@ -36,6 +41,7 @@ rb-api = { path = "src/rb-api", develop = true }
 pyinstaller = "^5.13.2"
 
 
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"

pyproject.toml

@@ -1,14 +1,9 @@
 from importlib.metadata import version
-
 import typer
 from rich import print
 
-from rescuebox.plugins import plugins
-
 app = typer.Typer()
-
 management_app = typer.Typer()
-
 __version__ = version("rescuebox")
 
 
@@ -27,6 +22,7 @@ def list_plugins() -> list[str]:
     """
     List all plugins
     """
+    from rescuebox.plugins import plugins  # Lazy Import
     print("Plugins:")
     plugin_list = []
     for plugin in plugins:
@@ -37,5 +33,14 @@ def list_plugins() -> list[str]:
 
 app.add_typer(management_app, name="manage")
 
+
+# Delay plugin loading to avoid circular import
+def load_plugins():
+    from rescuebox.plugins import plugins  # Import from __init__.py now
+    return plugins
+
+
+plugins = load_plugins()
+
 for plugin in plugins:
     app.add_typer(plugin.app, name=plugin.cli_name)

rescuebox/main.py

@@ -1,8 +1,10 @@
 from dataclasses import dataclass
-
 import typer
+
+# Import plugin modules
 from rb_doc_parser.main import app as rb_doc_parser_app  # type: ignore
 from rb_file_utils.main import app as rb_file_utils_app  # type: ignore
+from rb_audio_transcription.main import app as rb_audio_transcription_app  # type: ignore
 
 
 @dataclass(frozen=True)
@@ -12,8 +14,12 @@ class RescueBoxPlugin:
     full_name: str | None
 
 
-# Add plugins here
+# Define plugins here (NOT dynamically in main.py)
 plugins: list[RescueBoxPlugin] = [
     RescueBoxPlugin(rb_file_utils_app, "fs", "File Utils"),
     RescueBoxPlugin(rb_doc_parser_app, "docs", "Docs Utils"),
+    RescueBoxPlugin(rb_audio_transcription_app, "audio", "Audio transcription library"),
 ]
+
+# Ensure this module is importable
+__all__ = ["plugins"]