refactor: improve code formatting and readability

- Reformatted comments and docstrings in `screen_switch_tool.py` for better clarity.
- Cleaned up import statements in `askui_controller.py` for improved organization and readability.

Author: danyalxahid-askui

src/askui/tools/askui/askui_controller.py

@@ -7,7 +7,7 @@
 from typing import Literal, Type
 
 import grpc
-from google.protobuf.json_format import MessageToDict  # type: ignore
+from google.protobuf.json_format import MessageToDict
 from PIL import Image
 from pydantic import BaseModel, Field, model_validator
 from pydantic_settings import BaseSettings, SettingsConfigDict

src/askui/tools/screen_switch_tool.py

@@ -8,14 +8,16 @@ class ScreenSwitchTool(Tool):
     """
 
     def __init__(self, agent_os: AgentOs) -> None:
-        # We need to determine the number of displays available to provide context to the agent
-        # indicating that screen switching can only be done this number of times.
+        # We need to determine the number of displays available to provide context
+        # to the agent indicating that screen switching can only be done this number
+        # of times.
         displays: list[DisplayInformation] = agent_os.get_display_information().displays
 
         super().__init__(
             name="screen_switch",
             description=f"""
-            This tool is useful for switching between multiple displays to find information not present on the current active screen.
+            This tool is useful for switching between multiple displays to find
+            information not present on the current active screen.
             If more than one display is available, this tool cycles through them.
             Number of displays available: {len(displays)}.
             """,
@@ -26,7 +28,8 @@ def __init__(self, agent_os: AgentOs) -> None:
     def __call__(self) -> None:
         """
         Cycles to the next display if there are multiple displays.
-        This tool is useful to switch between multiple displays if some information is not found on the current display.
+        This tool is useful to switch between multiple displays if some information is
+        not found on the current display.
         """
         if len(self._displays) <= 1:
             return