naas-abi 1.0.0__py3-none-any.whl

naas_abi/apps/terminal_agent/terminal_style.py

@@ -0,0 +1,175 @@
+from rich.console import Console
+from rich.markdown import Markdown
+from rich.panel import Panel
+from rich.syntax import Syntax
+from rich.text import Text
+from rich.box import ROUNDED
+from PIL import Image
+import os
+import platform
+import subprocess
+
+console = Console()
+
+
+def set_terminal_title():
+    """Set the terminal title to 'ABI'"""
+    if platform.system() == "Windows":
+        os.system("title ABI")
+    else:  # For Unix-like systems (Linux, macOS)
+        print("\33]0;ABI\a", end="", flush=True)
+
+
+def print_agent_response(text, agent_label):
+    console.print()  # Add a blank line before the assistant's response
+    console.print(f"[bold green]{agent_label}:[/bold green] ", end="")
+    md = Markdown(text)
+    console.print(md)
+    console.print()  # Add a blank line after the assistant's response
+
+
+def print_system_message(text):
+    console.print()  # Add a blank line before the system message
+    system_text = Text(text, style="yellow")
+    console.print(
+        Panel(
+            system_text,
+            border_style="yellow",
+            box=ROUNDED,
+            expand=False,
+            title="System",
+            title_align="left",
+        )
+    )
+    console.print()  # Add a blank line after the system message
+
+
+def print_code(code, language="python"):
+    syntax = Syntax(code, language, theme="monokai", line_numbers=True)
+    console.print(
+        Panel(
+            syntax,
+            border_style="red",
+            box=ROUNDED,
+            expand=False,
+            title=f"Code ({language})",
+            title_align="left",
+        )
+    )
+
+
+def dict_to_equal_string(d: dict) -> str:
+    return "\n".join([f'-{key}="{value}"' for key, value in d.items()])
+
+
+def print_tool_usage(message):
+    print_message = ""
+    tool_name = message.tool_calls[0]["name"]
+    arguments = ""
+    if (
+        "args" in message.tool_calls[0]
+        and len(message.tool_calls[0]["args"].values()) > 0
+    ):
+        arguments += dict_to_equal_string(message.tool_calls[0]["args"])
+
+    if tool_name.startswith("transfer_to_"):
+        tool_label = " ".join(
+            word.capitalize()
+            for word in tool_name.split("transfer_to_")[1].replace("_", " ").split()
+        )
+        print_message = f"\n🧞 [bold blue]Delegated to [/bold blue]{tool_label}"
+    else:
+        tool_label = tool_name.capitalize().replace("_", " ")
+        print_message = f"\n[bold blue]Tool Used:[/bold blue] {tool_label}\n{arguments}"
+    console.print(print_message)
+
+
+def print_tool_response(response):
+    console.print(f"\n[bold blue]Response:[/bold blue] {response}")
+
+
+def clear_screen():
+    console.clear()
+
+
+def print_welcome_message(agent):
+    # Set terminal title
+    set_terminal_title()
+    
+    # Skip the welcome - we already said hello in the CLI startup
+    # Just quietly start the conversation
+    pass
+
+
+def print_divider():
+    console.print("─" * console.width, style="dim")
+
+
+def get_user_input(agent_label):
+    try:
+        user_input = console.input(
+            "\n[bold cyan]You:[/bold cyan] "
+        )  # Add a newline before the prompt
+        console.print()  # Add a blank line after the user's input
+        return user_input
+    except EOFError:
+        console.print(
+            f"\n[bold red]{agent_label}:[/bold red] Conversation ended by user."
+        )
+        return "exit"
+    except KeyboardInterrupt:
+        console.print(
+            f"\n[bold red]{agent_label}:[/bold red] Conversation ended by user."
+        )
+        return "exit"
+
+
+def print_image(image_path: str):
+    """Display an image in the terminal."""
+    try:
+        console.print()  # Add some spacing
+
+        # Display the file path and viewing instructions
+        message = (
+            f"[yellow]Image saved at: {image_path}[/yellow]\n\n"
+            "[dim]To view the image:[/dim]\n"
+            f"[cyan]• Local system: Open {image_path} with your image viewer[/cyan]\n"
+            "[cyan]• Remote/SSH: Download the file to your local machine to view[/cyan]"
+        )
+
+        console.print(
+            Panel(
+                message,
+                border_style="yellow",
+                box=ROUNDED,
+                expand=False,
+                title="Graph Output",
+                title_align="left",
+            )
+        )
+
+        # Only try to show the image if we're in a GUI environment
+        if os.environ.get("DISPLAY") and platform.system() != "Windows":
+            try:
+                img = Image.open(image_path)
+                img.show()
+            except Exception:
+                pass  # Silently fail if we can't display the image
+        elif platform.system() == "Windows":
+            try:
+                subprocess.run(['start', '', image_path], shell=True)  # Windows-specific file opening
+            except Exception:
+                pass
+
+        console.print()  # Add some spacing after
+    except Exception as e:
+        console.print(
+            Panel(
+                f"[yellow]Unable to process image. File saved at: {image_path}[/yellow]\nError: {str(e)}",
+                border_style="yellow",
+                box=ROUNDED,
+                expand=False,
+                title="Graph Output",
+                title_align="left",
+            )
+        )