PraisonAI 3.0.0__py3-none-any.whl

praisonai/cli/features/tui/screens/main.py

@@ -0,0 +1,245 @@
+"""
+Main Screen for PraisonAI TUI.
+
+The primary chat interface screen.
+"""
+
+from typing import Optional
+
+try:
+    from textual.screen import Screen
+    from textual.containers import Horizontal, Vertical, Container
+    from textual.widgets import Static, Footer, Header
+    from textual.binding import Binding
+    from textual.message import Message
+    TEXTUAL_AVAILABLE = True
+except ImportError:
+    TEXTUAL_AVAILABLE = False
+    Screen = object
+    Message = object
+
+
+if TEXTUAL_AVAILABLE:
+    from ..widgets.chat import ChatWidget, ChatMessage
+    from ..widgets.composer import ComposerWidget
+    from ..widgets.status import StatusWidget, StatusInfo
+    from ..widgets.queue_panel import QueuePanelWidget
+    from ..widgets.tool_panel import ToolPanelWidget
+
+    class MainScreen(Screen):
+        """
+        Main chat screen.
+        
+        Layout:
+        - Status bar (top)
+        - Chat history (main area)
+        - Tools panel (right sidebar)
+        - Queue panel (right sidebar)
+        - Composer (bottom)
+        - Footer with keybindings
+        """
+        
+        BINDINGS = [
+            Binding("ctrl+q", "quit", "Quit", show=True),
+            Binding("ctrl+c", "cancel", "Cancel", show=True),
+            Binding("ctrl+l", "clear", "Clear", show=True),
+            Binding("f1", "help", "Help", show=True),
+            Binding("f2", "toggle_queue", "Queue", show=True),
+            Binding("f3", "settings", "Settings", show=True),
+            Binding("f5", "clear_chat", "Clear Chat", show=True),
+            Binding("tab", "focus_next", "Next", show=False),
+            Binding("escape", "cancel_input", "Cancel", show=False),
+        ]
+        
+        DEFAULT_CSS = """
+        MainScreen {
+            layout: grid;
+            grid-size: 2;
+            grid-columns: 3fr 1fr;
+            grid-rows: 1 1fr auto 1;
+        }
+        
+        MainScreen #status-bar {
+            column-span: 2;
+        }
+        
+        MainScreen #main-area {
+            height: 100%;
+        }
+        
+        MainScreen #sidebar {
+            height: 100%;
+        }
+        
+        MainScreen #composer-area {
+            column-span: 2;
+            height: auto;
+            max-height: 10;
+        }
+        
+        MainScreen Footer {
+            column-span: 2;
+        }
+        
+        MainScreen .sidebar-panel {
+            height: 1fr;
+        }
+        """
+        
+        class MessageSubmitted(Message):
+            """Event when a message is submitted."""
+            def __init__(self, content: str):
+                self.content = content
+                super().__init__()
+        
+        class CommandExecuted(Message):
+            """Event when a command is executed."""
+            def __init__(self, command: str, args: str):
+                self.command = command
+                self.args = args
+                super().__init__()
+        
+        class CancelRequested(Message):
+            """Event when cancel is requested."""
+            pass
+        
+        def __init__(
+            self,
+            show_queue: bool = True,
+            show_tools: bool = True,
+            name: Optional[str] = None,
+            id: Optional[str] = None,
+            classes: Optional[str] = None,
+        ):
+            super().__init__(name=name, id=id, classes=classes)
+            self._show_queue = show_queue
+            self._show_tools = show_tools
+        
+        def compose(self):
+            """Compose the screen."""
+            yield StatusWidget(id="status-bar")
+            
+            yield ChatWidget(id="chat-widget")
+            
+            with Vertical(id="sidebar"):
+                if self._show_tools:
+                    yield ToolPanelWidget(id="tool-panel", classes="sidebar-panel")
+                if self._show_queue:
+                    yield QueuePanelWidget(id="queue-panel", classes="sidebar-panel")
+            
+            yield ComposerWidget(id="composer")
+            
+            yield Footer()
+        
+        def on_mount(self) -> None:
+            """Handle mount."""
+            # Focus composer
+            composer = self.query_one("#composer", ComposerWidget)
+            composer.focus_input()
+        
+        def on_composer_widget_submitted(self, event: ComposerWidget.Submitted) -> None:
+            """Handle message submission."""
+            if event.is_command:
+                # Parse command
+                parts = event.content[1:].split(maxsplit=1)
+                command = parts[0] if parts else ""
+                args = parts[1] if len(parts) > 1 else ""
+                self.post_message(self.CommandExecuted(command, args))
+            else:
+                self.post_message(self.MessageSubmitted(event.content))
+        
+        def action_quit(self) -> None:
+            """Quit the application."""
+            self.app.exit()
+        
+        def action_cancel(self) -> None:
+            """Cancel current operation."""
+            self.post_message(self.CancelRequested())
+        
+        def action_clear(self) -> None:
+            """Clear screen."""
+            pass
+        
+        def action_help(self) -> None:
+            """Show help."""
+            self.app.push_screen("help")
+        
+        def action_toggle_queue(self) -> None:
+            """Toggle queue panel visibility."""
+            try:
+                queue_panel = self.query_one("#queue-panel", QueuePanelWidget)
+                queue_panel.display = not queue_panel.display
+            except Exception:
+                pass
+        
+        def action_settings(self) -> None:
+            """Show settings."""
+            self.app.push_screen("settings")
+        
+        def action_clear_chat(self) -> None:
+            """Clear chat history."""
+            chat = self.query_one("#chat-widget", ChatWidget)
+            self.app.call_later(chat.clear)
+        
+        def action_cancel_input(self) -> None:
+            """Cancel current input."""
+            composer = self.query_one("#composer", ComposerWidget)
+            composer.action_cancel()
+        
+        # Public methods for updating UI
+        
+        async def add_user_message(self, content: str) -> None:
+            """Add a user message to chat."""
+            chat = self.query_one("#chat-widget", ChatWidget)
+            await chat.add_message(ChatMessage(
+                role="user",
+                content=content,
+            ))
+        
+        async def add_assistant_message(
+            self,
+            content: str,
+            run_id: Optional[str] = None,
+            agent_name: Optional[str] = None,
+            streaming: bool = False,
+        ) -> None:
+            """Add an assistant message to chat."""
+            chat = self.query_one("#chat-widget", ChatWidget)
+            await chat.add_message(ChatMessage(
+                role="assistant",
+                content=content,
+                run_id=run_id,
+                agent_name=agent_name,
+                is_streaming=streaming,
+            ))
+        
+        async def update_streaming(self, run_id: str, content: str) -> None:
+            """Update streaming message."""
+            chat = self.query_one("#chat-widget", ChatWidget)
+            await chat.update_streaming(run_id, content)
+        
+        async def complete_streaming(self, run_id: str, content: str) -> None:
+            """Complete streaming message."""
+            chat = self.query_one("#chat-widget", ChatWidget)
+            await chat.complete_streaming(run_id, content)
+        
+        def update_status(self, info: StatusInfo) -> None:
+            """Update status bar."""
+            status = self.query_one("#status-bar", StatusWidget)
+            status.update_info(info)
+        
+        def set_processing(self, processing: bool) -> None:
+            """Set processing state."""
+            composer = self.query_one("#composer", ComposerWidget)
+            composer.set_processing(processing)
+            
+            status = self.query_one("#status-bar", StatusWidget)
+            status.is_processing = processing
+
+else:
+    class MainScreen:
+        """Placeholder when Textual is not available."""
+        def __init__(self, *args, **kwargs):
+            raise ImportError(
+                "Textual is required for TUI. Install with: pip install praisonai[tui]"
+            )

praisonai/cli/features/tui/screens/queue.py

@@ -0,0 +1,174 @@
+"""
+Queue Screen for PraisonAI TUI.
+
+Queue management interface.
+"""
+
+from typing import Optional, List
+
+try:
+    from textual.screen import Screen
+    from textual.containers import Vertical, Horizontal
+    from textual.widgets import Static, Footer, DataTable, Button
+    from textual.binding import Binding
+    from textual.message import Message
+    from rich.text import Text
+    TEXTUAL_AVAILABLE = True
+except ImportError:
+    TEXTUAL_AVAILABLE = False
+    Screen = object
+    Message = object
+
+
+if TEXTUAL_AVAILABLE:
+    class QueueScreen(Screen):
+        """
+        Queue management screen.
+        
+        Displays full queue with management options.
+        """
+        
+        BINDINGS = [
+            Binding("escape", "back", "Back", show=True),
+            Binding("c", "cancel", "Cancel", show=True),
+            Binding("r", "retry", "Retry", show=True),
+            Binding("x", "clear", "Clear Queue", show=True),
+            Binding("enter", "select", "Select", show=False),
+        ]
+        
+        DEFAULT_CSS = """
+        QueueScreen {
+            background: $surface;
+        }
+        
+        QueueScreen #queue-header {
+            height: 3;
+            background: $primary;
+            padding: 1;
+        }
+        
+        QueueScreen DataTable {
+            height: 1fr;
+        }
+        
+        QueueScreen #queue-actions {
+            height: 3;
+            padding: 1;
+        }
+        """
+        
+        class RunCancelled(Message):
+            def __init__(self, run_id: str):
+                self.run_id = run_id
+                super().__init__()
+        
+        class RunRetried(Message):
+            def __init__(self, run_id: str):
+                self.run_id = run_id
+                super().__init__()
+        
+        class QueueCleared(Message):
+            pass
+        
+        def __init__(
+            self,
+            name: Optional[str] = None,
+            id: Optional[str] = None,
+            classes: Optional[str] = None,
+        ):
+            super().__init__(name=name, id=id, classes=classes)
+            self._selected_run_id: Optional[str] = None
+        
+        def compose(self):
+            """Compose the screen."""
+            yield Static("Queue Management", id="queue-header")
+            yield DataTable(id="queue-table")
+            with Horizontal(id="queue-actions"):
+                yield Button("Cancel [C]", id="btn-cancel", variant="error")
+                yield Button("Retry [R]", id="btn-retry", variant="warning")
+                yield Button("Clear [X]", id="btn-clear", variant="default")
+                yield Button("Back [Esc]", id="btn-back", variant="primary")
+            yield Footer()
+        
+        def on_mount(self) -> None:
+            """Handle mount."""
+            table = self.query_one("#queue-table", DataTable)
+            table.add_columns(
+                "ID", "Agent", "Input", "State", "Priority", 
+                "Wait Time", "Duration"
+            )
+            table.cursor_type = "row"
+        
+        def update_runs(self, runs: List[dict]) -> None:
+            """Update the runs display."""
+            table = self.query_one("#queue-table", DataTable)
+            table.clear()
+            
+            for run in runs:
+                state = run.get("state", "unknown")
+                state_style = {
+                    "queued": "yellow",
+                    "running": "green",
+                    "succeeded": "cyan",
+                    "failed": "red",
+                    "cancelled": "dim",
+                }.get(state, "")
+                
+                state_text = Text(state, style=state_style)
+                
+                input_preview = run.get("input", "")[:40]
+                if len(run.get("input", "")) > 40:
+                    input_preview += "..."
+                
+                table.add_row(
+                    run.get("run_id", "")[:8],
+                    run.get("agent_name", ""),
+                    input_preview,
+                    state_text,
+                    run.get("priority", "normal"),
+                    run.get("wait_time", "-"),
+                    run.get("duration", "-"),
+                    key=run.get("run_id"),
+                )
+        
+        def on_data_table_row_selected(self, event: DataTable.RowSelected) -> None:
+            """Handle row selection."""
+            if event.row_key:
+                self._selected_run_id = str(event.row_key.value)
+        
+        def on_button_pressed(self, event: Button.Pressed) -> None:
+            """Handle button press."""
+            if event.button.id == "btn-cancel":
+                self.action_cancel()
+            elif event.button.id == "btn-retry":
+                self.action_retry()
+            elif event.button.id == "btn-clear":
+                self.action_clear()
+            elif event.button.id == "btn-back":
+                self.action_back()
+        
+        def action_back(self) -> None:
+            """Go back to main screen."""
+            self.app.pop_screen()
+        
+        def action_cancel(self) -> None:
+            """Cancel selected run."""
+            if self._selected_run_id:
+                self.post_message(self.RunCancelled(self._selected_run_id))
+        
+        def action_retry(self) -> None:
+            """Retry selected run."""
+            if self._selected_run_id:
+                self.post_message(self.RunRetried(self._selected_run_id))
+        
+        def action_clear(self) -> None:
+            """Clear the queue."""
+            self.post_message(self.QueueCleared())
+
+else:
+    class QueueScreen:
+        """Placeholder when Textual is not available."""
+        def __init__(self, *args, **kwargs):
+            raise ImportError(
+                "Textual is required for TUI. Install with: pip install praisonai[tui]"
+            )

praisonai/cli/features/tui/screens/session.py

@@ -0,0 +1,124 @@
+"""
+Session Screen for PraisonAI TUI.
+
+Session browser and management.
+"""
+
+from typing import Optional, List
+
+try:
+    from textual.screen import Screen
+    from textual.containers import Vertical
+    from textual.widgets import Static, Footer, DataTable, Button
+    from textual.binding import Binding
+    from textual.message import Message
+    from rich.text import Text
+    TEXTUAL_AVAILABLE = True
+except ImportError:
+    TEXTUAL_AVAILABLE = False
+    Screen = object
+    Message = object
+
+
+if TEXTUAL_AVAILABLE:
+    class SessionScreen(Screen):
+        """
+        Session browser screen.
+        
+        Displays recent sessions and allows resuming.
+        """
+        
+        BINDINGS = [
+            Binding("escape", "back", "Back", show=True),
+            Binding("enter", "resume", "Resume", show=True),
+            Binding("d", "delete", "Delete", show=True),
+        ]
+        
+        DEFAULT_CSS = """
+        SessionScreen {
+            background: $surface;
+        }
+        
+        SessionScreen #session-header {
+            height: 3;
+            background: $primary;
+            padding: 1;
+        }
+        
+        SessionScreen DataTable {
+            height: 1fr;
+        }
+        """
+        
+        class SessionSelected(Message):
+            def __init__(self, session_id: str):
+                self.session_id = session_id
+                super().__init__()
+        
+        class SessionDeleted(Message):
+            def __init__(self, session_id: str):
+                self.session_id = session_id
+                super().__init__()
+        
+        def __init__(
+            self,
+            name: Optional[str] = None,
+            id: Optional[str] = None,
+            classes: Optional[str] = None,
+        ):
+            super().__init__(name=name, id=id, classes=classes)
+            self._selected_session_id: Optional[str] = None
+        
+        def compose(self):
+            """Compose the screen."""
+            yield Static("Sessions", id="session-header")
+            yield DataTable(id="session-table")
+            yield Footer()
+        
+        def on_mount(self) -> None:
+            """Handle mount."""
+            table = self.query_one("#session-table", DataTable)
+            table.add_columns("ID", "Created", "Updated", "Runs")
+            table.cursor_type = "row"
+        
+        def update_sessions(self, sessions: List[dict]) -> None:
+            """Update the sessions display."""
+            table = self.query_one("#session-table", DataTable)
+            table.clear()
+            
+            for session in sessions:
+                table.add_row(
+                    session.get("session_id", "")[:8],
+                    session.get("created_at", "-"),
+                    session.get("updated_at", "-"),
+                    str(session.get("run_count", 0)),
+                    key=session.get("session_id"),
+                )
+        
+        def on_data_table_row_selected(self, event: DataTable.RowSelected) -> None:
+            """Handle row selection."""
+            if event.row_key:
+                self._selected_session_id = str(event.row_key.value)
+        
+        def action_back(self) -> None:
+            """Go back."""
+            self.app.pop_screen()
+        
+        def action_resume(self) -> None:
+            """Resume selected session."""
+            if self._selected_session_id:
+                self.post_message(self.SessionSelected(self._selected_session_id))
+                self.app.pop_screen()
+        
+        def action_delete(self) -> None:
+            """Delete selected session."""
+            if self._selected_session_id:
+                self.post_message(self.SessionDeleted(self._selected_session_id))
+
+else:
+    class SessionScreen:
+        """Placeholder when Textual is not available."""
+        def __init__(self, *args, **kwargs):
+            raise ImportError(
+                "Textual is required for TUI. Install with: pip install praisonai[tui]"
+            )

praisonai/cli/features/tui/screens/settings.py

@@ -0,0 +1,148 @@
+"""
+Settings Screen for PraisonAI TUI.
+
+Configuration interface.
+"""
+
+from typing import Optional
+
+try:
+    from textual.screen import Screen
+    from textual.containers import Vertical
+    from textual.widgets import Static, Footer, Input, Select, Switch, Button
+    from textual.binding import Binding
+    from textual.message import Message
+    TEXTUAL_AVAILABLE = True
+except ImportError:
+    TEXTUAL_AVAILABLE = False
+    Screen = object
+    Message = object
+
+
+if TEXTUAL_AVAILABLE:
+    class SettingsScreen(Screen):
+        """
+        Settings screen.
+        
+        Allows configuration of:
+        - Model selection
+        - Queue settings
+        - Display preferences
+        """
+        
+        BINDINGS = [
+            Binding("escape", "back", "Back", show=True),
+            Binding("s", "save", "Save", show=True),
+        ]
+        
+        DEFAULT_CSS = """
+        SettingsScreen {
+            background: $surface;
+        }
+        
+        SettingsScreen #settings-header {
+            height: 3;
+            background: $primary;
+            padding: 1;
+        }
+        
+        SettingsScreen .setting-row {
+            height: 3;
+            padding: 0 1;
+        }
+        
+        SettingsScreen .setting-label {
+            width: 20;
+        }
+        
+        SettingsScreen Input {
+            width: 40;
+        }
+        """
+        
+        class SettingsSaved(Message):
+            def __init__(self, settings: dict):
+                self.settings = settings
+                super().__init__()
+        
+        def __init__(
+            self,
+            current_settings: Optional[dict] = None,
+            name: Optional[str] = None,
+            id: Optional[str] = None,
+            classes: Optional[str] = None,
+        ):
+            super().__init__(name=name, id=id, classes=classes)
+            self._settings = current_settings or {}
+        
+        def compose(self):
+            """Compose the screen."""
+            yield Static("Settings", id="settings-header")
+            
+            with Vertical(id="settings-form"):
+                # Model setting
+                yield Static("Model:", classes="setting-label")
+                yield Input(
+                    value=self._settings.get("model", "gpt-4o-mini"),
+                    id="setting-model",
+                    placeholder="Model name"
+                )
+                
+                # Max concurrent runs
+                yield Static("Max Concurrent:", classes="setting-label")
+                yield Input(
+                    value=str(self._settings.get("max_concurrent", 4)),
+                    id="setting-max-concurrent",
+                    placeholder="4"
+                )
+                
+                # Auto-save interval
+                yield Static("Autosave (sec):", classes="setting-label")
+                yield Input(
+                    value=str(self._settings.get("autosave_interval", 30)),
+                    id="setting-autosave",
+                    placeholder="30"
+                )
+                
+                # Buttons
+                yield Button("Save", id="btn-save", variant="primary")
+                yield Button("Cancel", id="btn-cancel", variant="default")
+            
+            yield Footer()
+        
+        def on_button_pressed(self, event: Button.Pressed) -> None:
+            """Handle button press."""
+            if event.button.id == "btn-save":
+                self.action_save()
+            elif event.button.id == "btn-cancel":
+                self.action_back()
+        
+        def action_back(self) -> None:
+            """Go back."""
+            self.app.pop_screen()
+        
+        def action_save(self) -> None:
+            """Save settings."""
+            try:
+                model_input = self.query_one("#setting-model", Input)
+                max_concurrent_input = self.query_one("#setting-max-concurrent", Input)
+                autosave_input = self.query_one("#setting-autosave", Input)
+                
+                settings = {
+                    "model": model_input.value,
+                    "max_concurrent": int(max_concurrent_input.value or "4"),
+                    "autosave_interval": int(autosave_input.value or "30"),
+                }
+                
+                self.post_message(self.SettingsSaved(settings))
+                self.app.pop_screen()
+            except ValueError:
+                pass
+
+else:
+    class SettingsScreen:
+        """Placeholder when Textual is not available."""
+        def __init__(self, *args, **kwargs):
+            raise ImportError(
+                "Textual is required for TUI. Install with: pip install praisonai[tui]"
+            )