aline-ai 0.7.2__py3-none-any.whl → 0.7.4__py3-none-any.whl

realign/dashboard/screens/create_agent_info.py

@@ -7,7 +7,7 @@ from typing import Optional
 
 from textual.app import ComposeResult
 from textual.binding import Binding
-from textual.containers import Container, Horizontal, Vertical
+from textual.containers import Container, Horizontal
 from textual.screen import ModalScreen
 from textual.widgets import Button, Input, Label, Static
 from textual.worker import Worker, WorkerState
@@ -34,7 +34,8 @@ class CreateAgentInfoScreen(ModalScreen[Optional[dict]]):
     }
 
     CreateAgentInfoScreen #create-agent-info-root {
-        width: 65;
+        width: 100%;
+        max-width: 65;
         height: auto;
         max-height: 80%;
         padding: 1 2;
@@ -57,15 +58,6 @@ class CreateAgentInfoScreen(ModalScreen[Optional[dict]]):
 
     CreateAgentInfoScreen Input {
         margin-top: 0;
-        border: none;
-    }
-
-    CreateAgentInfoScreen #or-separator {
-        height: auto;
-        margin-top: 1;
-        margin-bottom: 0;
-        text-align: center;
-        color: $text-muted;
     }
 
     CreateAgentInfoScreen #import-status {
@@ -74,23 +66,13 @@ class CreateAgentInfoScreen(ModalScreen[Optional[dict]]):
         color: $text-muted;
     }
 
-    CreateAgentInfoScreen #create-buttons {
+    CreateAgentInfoScreen #add-buttons {
         height: auto;
         margin-top: 1;
         align: right middle;
     }
 
-    CreateAgentInfoScreen #import-buttons {
-        height: auto;
-        margin-top: 1;
-        align: right middle;
-    }
-
-    CreateAgentInfoScreen #create-buttons Button {
-        margin-left: 1;
-    }
-
-    CreateAgentInfoScreen #import-buttons Button {
+    CreateAgentInfoScreen #add-buttons Button {
         margin-left: 1;
     }
     """
@@ -103,44 +85,37 @@ class CreateAgentInfoScreen(ModalScreen[Optional[dict]]):
 
     def compose(self) -> ComposeResult:
         with Container(id="create-agent-info-root"):
-            yield Static("Create Agent Profile", id="create-agent-info-title")
-
-            # --- Create New section ---
-            yield Label("Name", classes="section-label")
-            yield Input(placeholder=self._default_name, id="agent-name")
+            yield Static("Add Agent", id="create-agent-info-title")
 
-            with Horizontal(id="create-buttons"):
-                yield Button("Cancel", id="cancel")
-                yield Button("Create", id="create", variant="primary")
-
-            # --- Separator ---
-            yield Static("-- Or --", id="or-separator")
-
-            # --- Import from Link section ---
-            yield Label("Import from Link", classes="section-label")
-            yield Input(placeholder="https://realign-server.vercel.app/share/...", id="share-url")
-
-            yield Label("Password (optional)", classes="section-label")
-            yield Input(placeholder="Leave blank if not password-protected", id="share-password", password=True)
+            yield Label("Name or Share Link", classes="section-label")
+            yield Input(placeholder=self._default_name, id="agent-input")
 
             yield Static("", id="import-status")
 
-            with Horizontal(id="import-buttons"):
-                yield Button("Import", id="import", variant="primary")
+            with Horizontal(id="add-buttons"):
+                yield Button("Cancel", id="cancel")
+                yield Button("Add", id="add", variant="primary")
 
     def on_mount(self) -> None:
-        self.query_one("#agent-name", Input).focus()
+        agent_input = self.query_one("#agent-input", Input)
+        agent_input.styles.border = ("round", "black")
+        agent_input.focus()
+        for btn in (self.query_one("#cancel", Button), self.query_one("#add", Button)):
+            btn.styles.border = ("round", "black")
+            btn.styles.text_align = "center"
+            btn.styles.content_align = ("center", "middle")
 
     def action_close(self) -> None:
         self.dismiss(None)
 
     def _set_busy(self, busy: bool) -> None:
-        self.query_one("#agent-name", Input).disabled = busy
-        self.query_one("#share-url", Input).disabled = busy
-        self.query_one("#share-password", Input).disabled = busy
-        self.query_one("#create", Button).disabled = busy
+        self.query_one("#agent-input", Input).disabled = busy
+        self.query_one("#add", Button).disabled = busy
         self.query_one("#cancel", Button).disabled = busy
-        self.query_one("#import", Button).disabled = busy
+
+    def _is_share_link(self, value: str) -> bool:
+        """Check if the input looks like a share link."""
+        return "/share/" in value
 
     async def on_button_pressed(self, event: Button.Pressed) -> None:
         button_id = event.button.id or ""
@@ -149,28 +124,29 @@ class CreateAgentInfoScreen(ModalScreen[Optional[dict]]):
             self.dismiss(None)
             return
 
-        if button_id == "create":
-            await self._create_agent()
-            return
-
-        if button_id == "import":
-            await self._import_agent()
+        if button_id == "add":
+            await self._add_agent()
             return
 
     async def on_input_submitted(self, event: Input.Submitted) -> None:
         """Handle enter key in input fields."""
-        input_id = event.input.id or ""
-        if input_id == "agent-name":
-            await self._create_agent()
-        elif input_id in ("share-url", "share-password"):
-            await self._import_agent()
+        if event.input.id == "agent-input":
+            await self._add_agent()
 
-    async def _create_agent(self) -> None:
+    async def _add_agent(self) -> None:
+        """Create a new agent or import from link based on input content."""
+        value = self.query_one("#agent-input", Input).value.strip()
+
+        if self._is_share_link(value):
+            await self._import_agent(value)
+        else:
+            await self._create_agent(value)
+
+    async def _create_agent(self, name_input: str) -> None:
         """Create the agent profile."""
         try:
             from ...db import get_database
 
-            name_input = self.query_one("#agent-name", Input).value.strip()
             name = name_input or self._default_name
 
             agent_id = str(uuid.uuid4())
@@ -187,21 +163,8 @@ class CreateAgentInfoScreen(ModalScreen[Optional[dict]]):
             logger.error(f"Failed to create agent: {e}")
             self.app.notify(f"Failed to create agent: {e}", severity="error")
 
-    async def _import_agent(self) -> None:
+    async def _import_agent(self, share_url: str) -> None:
         """Import an agent from a share link."""
-        share_url = self.query_one("#share-url", Input).value.strip()
-        password = self.query_one("#share-password", Input).value.strip() or None
-
-        if not share_url:
-            self.app.notify("Please enter a share URL", severity="warning")
-            self.query_one("#share-url", Input).focus()
-            return
-
-        if "/share/" not in share_url:
-            self.app.notify("Invalid share URL format", severity="warning")
-            self.query_one("#share-url", Input).focus()
-            return
-
         status = self.query_one("#import-status", Static)
         status.update("Importing...")
         self._set_busy(True)
@@ -209,7 +172,7 @@ class CreateAgentInfoScreen(ModalScreen[Optional[dict]]):
         def do_import() -> dict:
             from ...commands.import_shares import import_agent_from_share
 
-            return import_agent_from_share(share_url, password=password)
+            return import_agent_from_share(share_url, password=None)
 
         self._import_worker = self.run_worker(do_import, thread=True, exit_on_error=False)