Create routing.py

app/services/routing.py

@@ -0,0 +1,86 @@
+# app/services/routing.py
+from fastapi import APIRouter
+from typing import Dict, Any, Optional
+
+from app.configs import CLASSIFIER_URI, SEGMENTER_URI
+
+
+router = APIRouter()
+
+# =========================
+# Model Registry (for A/B testing)
+# =========================
+# Maps version names to actual model URIs
+MODEL_REGISTRY = {
+    "default": CLASSIFIER_URI,
+    "v1": CLASSIFIER_URI,
+    "hf_savedmodel": "hf_savedmodel",
+    # يمكنك إضافة إصدارات أخرى هنا
+    # "v2": "hf_savedmodel_v2",
+}
+
+# A/B Testing configuration
+AB_TESTING_ENABLED = False
+AB_TESTING_RATIO = 0.0  # 0.0 = no traffic to new model
+
+
+def get_ab_status() -> Dict[str, Any]:
+    """Get current A/B testing status"""
+    return {
+        "enabled": AB_TESTING_ENABLED,
+        "ratio": AB_TESTING_RATIO,
+        "default_version": CLASSIFIER_URI,
+        "available_versions": list(MODEL_REGISTRY.keys()),
+    }
+
+
+def resolve_classifier_version(version: Optional[str]) -> str:
+    """
+    Resolve a version name to actual model URI.
+    
+    Args:
+        version: Version name (e.g., "v1", "default") or None
+    
+    Returns:
+        Actual model URI
+    """
+    if version is None or version == "":
+        return CLASSIFIER_URI
+    
+    resolved = MODEL_REGISTRY.get(version)
+    if resolved is None:
+        # Fallback to default if version not found
+        return CLASSIFIER_URI
+    
+    return resolved
+
+
+# =========================
+# Optional: A/B Testing Router Endpoints
+# =========================
+@router.get("/api/models/versions")
+async def list_model_versions():
+    """List all available model versions"""
+    return {
+        "versions": list(MODEL_REGISTRY.keys()),
+        "default": CLASSIFIER_URI,
+        "registry": MODEL_REGISTRY,
+    }
+
+
+@router.post("/api/ab-testing/config")
+async def update_ab_config(enabled: bool = False, ratio: float = 0.0):
+    """Update A/B testing configuration (admin only)"""
+    global AB_TESTING_ENABLED, AB_TESTING_RATIO
+    
+    if not 0.0 <= ratio <= 1.0:
+        return {"error": "Ratio must be between 0.0 and 1.0"}
+    
+    AB_TESTING_ENABLED = enabled
+    AB_TESTING_RATIO = ratio
+    
+    return {
+        "status": "updated",
+        "enabled": AB_TESTING_ENABLED,
+        "ratio": AB_TESTING_RATIO,
+    }