Initial commit: FunASR Speech Recognition Toolkit

Add complete FunASR codebase including models, runtime, and documentation.

funasr/__init__.py

@@ -0,0 +1,79 @@
+"""Initialize funasr package."""
+
+import importlib
+import os
+import pkgutil
+import traceback
+
+
+dirname = os.path.dirname(__file__)
+version_file = os.path.join(dirname, "version.txt")
+with open(version_file, "r") as f:
+    __version__ = f.read().strip()
+
+
+_IMPORT_ERRORS = {}
+_IMPORT_ERROR_TRACEBACKS = {}
+_IMPORT_DEBUG = os.environ.get("FUNASR_IMPORT_DEBUG") == "1"
+_STRICT_IMPORT = os.environ.get("FUNASR_STRICT_IMPORT") == "1"
+
+
+def _record_import_error(name, error):
+    """Internal: record import error.
+    
+        Args:
+            name: TODO.
+            error: TODO.
+        """
+    _IMPORT_ERRORS[name] = f"{error.__class__.__name__}: {error}"
+    _IMPORT_ERROR_TRACEBACKS[name] = traceback.format_exc()
+    if _IMPORT_DEBUG:
+        print(f"Failed to import {name}: {_IMPORT_ERRORS[name]}")
+
+
+def get_import_errors():
+    """Get import errors."""
+    return dict(_IMPORT_ERRORS)
+
+
+def get_import_error_tracebacks():
+    """Get import error tracebacks."""
+    return dict(_IMPORT_ERROR_TRACEBACKS)
+
+
+def import_submodules(package, recursive=True):
+    """Import submodules.
+    
+        Args:
+            package: TODO.
+            recursive: TODO.
+        """
+    if isinstance(package, str):
+        try:
+            package = importlib.import_module(package)
+        except Exception as e:
+            _record_import_error(package, e)
+            if _STRICT_IMPORT:
+                raise
+            return {}
+    results = {}
+    if not isinstance(package, str):
+        for loader, name, is_pkg in pkgutil.walk_packages(package.__path__, package.__name__ + "."):
+            try:
+                results[name] = importlib.import_module(name)
+            except Exception as e:
+                _record_import_error(name, e)
+                if _STRICT_IMPORT:
+                    raise
+                continue
+            if recursive and is_pkg:
+                results.update(import_submodules(name))
+    return results
+
+
+import_submodules(__name__)
+
+from funasr.auto.auto_model import AutoModel
+from funasr.auto.auto_frontend import AutoFrontend
+
+os.environ["HYDRA_FULL_ERROR"] = "1"