"""
LegalSync Orchestrator Agent — main entry point.

Validated against the DataRobot Agentic AI framework (`MyAgent.invoke`
pattern): the agent wires the sub-processors and exposes a workflow router
consumed by workload_app.py. Runs over the DataRobot Workload API (any HTTP
container) with Groq / DataRobot LLM Gateway for any model calls (00-core).
"""
from services.omnipay_payment_processor import OmniPayProcessor
from services.oracle_trust_fund_processor import TrustFundProcessor
from services.whisper_bodycam_transcriber import BodycamTranscriber
from services.discovery_assembler import DiscoveryAssembler
from services.bond_recommendation_agent import BondRecommendationAgent


class LegalSyncAgent:
    def __init__(self):
        self.payments = OmniPayProcessor()
        self.trust = TrustFundProcessor()
        self.transcriber = BodycamTranscriber()
        self.discovery = DiscoveryAssembler()
        self.bond = BondRecommendationAgent()

    def run_workflow(self, workflow_type, payload):
        """Workflow router (MyAgent.invoke pattern)."""
        if workflow_type == "payment":
            return self.payments.process(**payload)
        elif workflow_type == "trust-fund":
            return self.trust.get_balance(payload.get("inmate_id"))
        elif workflow_type == "transcription":
            return self.transcriber.transcribe(payload.get("video_path"))
        elif workflow_type == "discovery":
            return self.discovery.assemble(payload.get("case_id"), payload.get("transcripts", []), payload.get("bodycams", []))
        elif workflow_type == "bond":
            return self.bond.recommend(payload)
        return {"status": "unknown workflow", "supported": ["payment", "trust-fund", "transcription", "discovery", "bond"]}