Fix the cursor logo placement for ubuntu

desktop_env/server/main.py

@@ -3,7 +3,7 @@ from pathlib import Path
 import platform
 import subprocess
 import requests
-
+from .pyxcursor import Xcursor
 # import Xlib.display
 import pyautogui
 # from PIL import ImageGrab, Image
@@ -48,7 +48,7 @@ def capture_screen_with_cursor():
     os.makedirs(os.path.dirname(file_path), exist_ok=True)
 
     # fixme: This is a temporary fix for the cursor not being captured on Windows and Linux
-    if user_platform == "Windows" or user_platform == "Linux":
+    if user_platform == "Windows":
         def _download_image(url, path):
             response = requests.get(url)
             with open(path, 'wb') as file:
@@ -65,12 +65,14 @@ def capture_screen_with_cursor():
         cursor = cursor.resize((int(cursor.width / 1.5), int(cursor.height / 1.5)))
         screenshot.paste(cursor, (cursor_x, cursor_y), cursor)
         screenshot.save(file_path)
-    # elif user_platform == "Linux":
-    #     # Use xlib to prevent scrot dependency for Linux
-    #     screen = Xlib.display.Display().screen()
-    #     size = screen.width_in_pixels, screen.height_in_pixels
-    #     screenshot = ImageGrab.grab(bbox=(0, 0, size[0], size[1]))
-    #     screenshot.save(file_path)
+    elif user_platform == "Linux":
+        cursor_obj = Xcursor()
+        imgarray = cursor_obj.getCursorImageArrayFast()
+        cursor_img = Image.fromarray(imgarray)
+        screenshot = pyautogui.screenshot()
+        cursor_x, cursor_y = pyautogui.position()
+        screenshot.paste(cursor_img, (cursor_x, cursor_y), cursor_img)
+        screenshot.save(file_path)
     elif user_platform == "Darwin":  # (Mac OS)
         # Use the screencapture utility to capture the screen with the cursor
         subprocess.run(["screencapture", "-C", file_path])