#!/usr/bin/env python3
"""
Vytre Core Inference Script
Simple interface to run Vytre Core
"""

import json
import sys
from typing import Dict, Any


class VytreCoreInference:
    def __init__(self):
        self.system_prompt = """You are Vytre, an enterprise workforce operating intelligence model.
Your responsibilities:
- Understand organizational structures
- Create and manage AI agents
- Coordinate workforce execution
- Build workflows
- Delegate tasks
- Maintain governance
- Route tool actions
- Assist strategic execution

Optimize for:
- Enterprise operations
- Agent orchestration
- Workflow planning
- Task execution
- Organizational reasoning
"""

    def format_input(self, user_input: str) -> str:
        return f"""{self.system_prompt}
Input: {user_input}
Output: """

    def mock_predict(self, user_input: str) -> Dict[str, Any]:
        """
        Mock prediction function (replace with actual model inference when trained)
        """
        # Simple mock logic for demo
        if "department" in user_input.lower() or "organization" in user_input.lower():
            # Try to detect which department
            dept_name = "Marketing"
            dept_manager = "Marketing Manager"
            workers = ["SEO Agent", "Content Agent", "Analytics Agent"]
            
            if "sales" in user_input.lower():
                dept_name = "Sales"
                dept_manager = "Sales Manager"
                workers = ["Sales Agent", "Account Manager", "BDR Agent"]
            elif "support" in user_input.lower():
                dept_name = "Support"
                dept_manager = "Support Manager"
                workers = ["Support Agent", "Tier 2 Support Agent", "Escalation Agent"]
            elif "engineering" in user_input.lower() or "tech" in user_input.lower():
                dept_name = "Engineering"
                dept_manager = "Engineering Manager"
                workers = ["Backend Agent", "Frontend Agent", "DevOps Agent", "QA Agent"]
            elif "finance" in user_input.lower():
                dept_name = "Finance"
                dept_manager = "Finance Manager"
                workers = ["Accounting Agent", "Budget Agent"]
            elif "hr" in user_input.lower() or "human" in user_input.lower():
                dept_name = "HR"
                dept_manager = "HR Manager"
                workers = ["Recruiting Agent", "Onboarding Agent"]
            
            return {
                "objective": user_input,
                "departments": [
                    {
                        "name": dept_name,
                        "manager": dept_manager,
                        "workers": workers
                    }
                ],
                "approval_required": False
            }
        elif "agent" in user_input.lower():
            return {
                "name": "Customer Support Agent",
                "skills": ["Ticket analysis", "Customer response", "Escalation"],
                "permissions": ["read_customer_records"]
            }
        elif "workflow" in user_input.lower():
            return {
                "workflow": [
                    "Create account",
                    "Assign onboarding agent",
                    "Send welcome email",
                    "Schedule meeting"
                ]
            }
        elif "price" in user_input.lower() or "approval" in user_input.lower():
            return {
                "requires_approval": True,
                "reason": "Pricing changes affect customers and require executive approval"
            }
        else:
            return {
                "objective": user_input,
                "message": "Processing request..."
            }


def main():
    print("=== Vytre Core Inference ===")
    vytre = VytreCoreInference()
    
    if len(sys.argv) > 1:
        user_input = " ".join(sys.argv[1:])
        print(f"\nInput: {user_input}")
        result = vytre.mock_predict(user_input)
        print(f"\nOutput: {json.dumps(result, indent=2)}")
    else:
        # Interactive mode
        print("\nEnter 'quit' to exit")
        while True:
            try:
                user_input = input("\nEnter your request: ").strip()
                if user_input.lower() in ["quit", "exit", "q"]:
                    print("Goodbye!")
                    break
                if not user_input:
                    continue
                result = vytre.mock_predict(user_input)
                print(f"\nVytre Output: {json.dumps(result, indent=2)}")
            except KeyboardInterrupt:
                print("\nGoodbye!")
                break


if __name__ == "__main__":
    main()