42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from typing import List
|
|
|
|
from textual.app import ComposeResult
|
|
from textual.screen import Screen
|
|
|
|
from models.agent import Agent
|
|
from widgets.multiagentselector import MultiAgentSelector
|
|
from widgets.OTP_generate import OTPGenerator
|
|
|
|
|
|
class OTPWorkflowScreen(Screen):
|
|
"""Screen that handles the OTP generation workflow."""
|
|
|
|
def __init__(self, all_agents: List[Agent]):
|
|
super().__init__()
|
|
self.all_agents = all_agents
|
|
self.selected_devices = None
|
|
|
|
def compose(self) -> ComposeResult:
|
|
"""Start with the multi-agent selector."""
|
|
yield MultiAgentSelector(self.all_agents)
|
|
|
|
def on_multi_agent_selector_agents_selected(
|
|
self, message: MultiAgentSelector.AgentsSelected
|
|
) -> None:
|
|
"""Handle selected agents - switch to OTP generator."""
|
|
self.selected_devices = message.selected_agents
|
|
|
|
# Remove the MultiAgentSelector
|
|
selector = self.query_one(MultiAgentSelector)
|
|
selector.remove()
|
|
|
|
# Mount the OTPGenerator with the selected Agent objects
|
|
# No need to pass API - it will access self.app.api directly
|
|
self.mount(OTPGenerator(self.selected_devices))
|
|
|
|
def on_otp_generator_otp_info(self, message: OTPGenerator.OTPInfo) -> None:
|
|
"""Handle OTP generation request - call the actual OTP generation function."""
|
|
# This will be handled by the main app, but we can also do it here
|
|
# For now, just pass it up to the app level
|
|
pass
|