Upload 327 files

database/__init__.py

@@ -10,12 +10,18 @@ new manager if that module is unavailable.
 
 from importlib import util as _importlib_util
 from pathlib import Path as _Path
-from typing import Optional as _Optional
+from typing import Optional as _Optional, Any as _Any
 
 from .db_manager import DatabaseManager
 
-def _load_legacy_database() -> _Optional[type]:
-    """Load the legacy Database class from the root-level ``database.py`` if it exists."""
+
+def _load_legacy_module():
+    """Load the legacy root-level ``database.py`` module if it exists.
+
+    This is used to support older entry points like ``get_database`` and the
+    ``Database`` class that live in the legacy file.
+    """
+
     legacy_path = _Path(__file__).resolve().parent.parent / "database.py"
     if not legacy_path.exists():
         return None
@@ -26,19 +32,64 @@ def _load_legacy_database() -> _Optional[type]:
 
     module = _importlib_util.module_from_spec(spec)
     try:
-        spec.loader.exec_module(module)
+        spec.loader.exec_module(module)  # type: ignore[union-attr]
     except Exception:
         # If loading the legacy module fails we silently fall back to DatabaseManager
         return None
 
+    return module
+
+
+def _load_legacy_database_class() -> _Optional[type]:
+    """Load the legacy ``Database`` class from ``database.py`` if available."""
+
+    module = _load_legacy_module()
+    if module is None:
+        return None
     return getattr(module, "Database", None)
 
 
-_LegacyDatabase = _load_legacy_database()
+def _load_legacy_get_database() -> _Optional[callable]:
+    """Load the legacy ``get_database`` function from ``database.py`` if available."""
+
+    module = _load_legacy_module()
+    if module is None:
+        return None
+    return getattr(module, "get_database", None)
+
+
+_LegacyDatabase = _load_legacy_database_class()
+_LegacyGetDatabase = _load_legacy_get_database()
+_db_manager_instance: _Optional[DatabaseManager] = None
+
 
 if _LegacyDatabase is not None:
     Database = _LegacyDatabase
 else:
     Database = DatabaseManager
 
-__all__ = ["DatabaseManager", "Database"]
+
+def get_database(*args: _Any, **kwargs: _Any) -> _Any:
+    """Return a database instance compatible with legacy callers.
+
+    The resolution order is:
+
+    1. If the legacy ``database.py`` file exists and exposes ``get_database``,
+       use that function (this returns the legacy singleton used by the
+       Gradio crypto dashboard and other older modules).
+    2. Otherwise, return a singleton instance of ``DatabaseManager`` from the
+       new SQLAlchemy-backed implementation.
+    """
+
+    if _LegacyGetDatabase is not None:
+        return _LegacyGetDatabase(*args, **kwargs)
+
+    global _db_manager_instance
+    if _db_manager_instance is None:
+        _db_manager_instance = DatabaseManager()
+        # Ensure tables are created for the monitoring schema
+        _db_manager_instance.init_database()
+    return _db_manager_instance
+
+
+__all__ = ["DatabaseManager", "Database", "get_database"]

requirements.txt

@@ -1,9 +1,18 @@
+"""Unified dependencies for API server + Gradio dashboards (Hugging Face Spaces).
+
+This single file is what Hugging Face Spaces will install. It includes:
+- Core API server stack
+- Database/SQLAlchemy layer
+- Gradio dashboard + Plotly charts
+- Optional Transformers-based AI features
+"""
+
 # Core API Server Requirements
 fastapi==0.109.0
 uvicorn[standard]==0.27.0
 pydantic==2.5.3
 sqlalchemy==2.0.25
-httpx>=0.26.0
+httpx==0.26.0
 websockets>=12.0
 python-dotenv
 python-multipart
@@ -13,8 +22,12 @@ aiohttp>=3.8.0
 # Data Processing
 pandas>=2.1.0
 
-# For Gradio Dashboard (see requirements_gradio.txt for full list)
-# gradio>=4.12.0
-# plotly>=5.18.0
-# transformers>=4.36.0  # Optional: for AI features
-# torch>=2.0.0  # Optional: for AI features
+# Gradio Dashboard & UI
+gradio==4.12.0
+plotly==5.18.0        # Enables chart features in the dashboard
+psutil==5.9.6         # Optional, used for system monitoring widgets
+
+# AI/ML Libraries (optional but recommended for AI features)
+transformers>=4.36.0  # Used by ai_models and HF integration
+torch>=2.0.0          # Backend for transformers (CPU build on HF Spaces)
+sentencepiece>=0.1.99 # Required by some Transformer text models