import asyncio
import logging
import traceback
import os
from typing import Optional, List

# Native ADK Imports
try:
    from google.adk.agents import LiveRequestQueue, LlmAgent
    from google.adk.runners import Runner
    from google.genai import types
except ImportError:
    logging.error("CRITICAL: google-adk not installed.")
    pass

logger = logging.getLogger(__name__)

class NativeVoiceClient:
    """
    Antigravity Native Voice Client (2026).
    Powered by Google ADK (LlmAgent + LiveRequestQueue) & Gemini 3 Flash.
    """

    def __init__(self, api_key: str, model: str):
        self.model = model
        self.api_key = api_key
        self.runner = None
        self.is_running = False
        
        # Configure Native Toolkit
        self.tools = [
            types.Tool(
                function_declarations=[
                     types.FunctionDeclaration(
                         name="auto_browse",
                         description="Launch a browser to see and interact with websites.",
                         parameters=types.Schema(
                             type="OBJECT",
                             properties={
                                 "url": types.Schema(type="STRING", description="URL to visit"),
                                 "action": types.Schema(type="STRING", description="Action to perform")
                             }
                         )
                     )
                ]
            )
        ]

    async def start(self):
        """Ignite the Native Voice Bridge."""
        logger.info(f"Igniting Native Voice Bridge ({self.model})...")
        self.is_running = True
        
        try:
            # 1. Initialize Agent
            # Using LlmAgent with direct configuration. 
            # If 'model' kwarg is not accepted, we might need a Config object, but 'model' is standard.
            # Using 'model' to match usage.
            agent = LlmAgent(
                model=self.model,
                tools=self.tools
                # system_instruction?
            )
            
            # 2. Initialize Runner
            self.runner = Runner(
                app_name="antigravity_mentor",
                agent=agent,
                # session_service? defaults to memory
            )

            # 3. Initialize Live Request Queue
            live_queue = LiveRequestQueue()
            
            logger.info("<< NATIVE VOICE VISUAL SYNC ESTABLISHED >>")
            logger.info(f"Bridge Active for Model: {self.model}")
            
            # 4. Run Live
            # Passing live_request_queue as required by run_live
            async for event in self.runner.run_live(live_request_queue=live_queue):
                 # Process events if needed, or simple iteration to keep it running
                 pass

        except Exception as e:
            logger.error(f"Native Voice Bridge Failure: {e}")
            logger.debug(traceback.format_exc())
            
        finally:
            self.is_running = False
            logger.info("Native Voice Bridge Terminated.")

    async def stop(self):
        self.is_running = False
        if self.runner:
            # Stop mechanism if available, otherwise loop break in start() handles it via is_running (if we stored context)
            # ADK Runner run_live returns a generator, typically stopping iteration stops it?
            # Or we cancel the task running start()
            pass