computer-use-ootb-internal 0.0.171__py3-none-any.whl → 0.0.172__py3-none-any.whl

computer_use_ootb_internal/preparation/powerpoint_prepare.py

@@ -0,0 +1,91 @@
+import os
+import platform
+import subprocess
+import logging
+from pathlib import Path
+
+log = logging.getLogger(__name__)
+
+def run_preparation(state):
+    """
+    Performs environment preparation specific to PowerPoint.
+    Opens a specific template file located on the user's desktop.
+    """
+    system_platform = platform.system()
+    log.info(f"PowerPoint preparation: Starting on {system_platform} platform...")
+
+    try:
+        # Determine the desktop path based on platform
+        if system_platform == "Windows":
+            try:
+                # Get the Windows username from environment variables
+                username = os.environ.get("USERNAME", "")
+                if not username:
+                    log.error("Could not determine Windows username from environment")
+                    return
+                
+                log.info(f"Using Windows username: {username}")
+                # Construct path to user's desktop
+                desktop_path = Path(f"C:/Users/{username}/Desktop")
+                
+                # Check if path exists
+                if not desktop_path.exists():
+                    log.error(f"Desktop path not found at: {desktop_path}")
+                    # Try alternative locations if needed
+                    alt_path = Path(f"C:/Documents and Settings/{username}/Desktop")
+                    if alt_path.exists():
+                        desktop_path = alt_path
+                        log.info(f"Using alternative desktop path: {desktop_path}")
+                    else:
+                        log.error("Failed to find user's desktop directory")
+                        return
+                
+            except Exception as e:
+                log.error(f"Error determining Windows user desktop: {e}", exc_info=True)
+                return
+                
+        elif system_platform == "Darwin":  # macOS
+            # On macOS we can use a simpler approach
+            desktop_path = Path.home() / "Desktop"
+            log.info(f"Using macOS desktop path: {desktop_path}")
+        else:
+            log.warning(f"Platform {system_platform} not specifically supported. Attempting generic approach.")
+            desktop_path = Path.home() / "Desktop"
+            
+        # Construct path to template file
+        template_file = desktop_path / "template.txt"
+        log.info(f"Looking for template file at: {template_file}")
+
+        if not template_file.exists():
+            log.error(f"Template file not found at: {template_file}")
+            return
+
+        # Open the file with appropriate command based on platform
+        if system_platform == "Windows":
+            log.info(f"Attempting to open {template_file} with PowerPoint on Windows...")
+            try:
+                # Use start command on Windows
+                cmd = ['cmd', '/c', 'start', 'powerpnt', str(template_file)]
+                result = subprocess.run(cmd, check=False, capture_output=True, text=True)
+                
+                if result.returncode == 0:
+                    log.info(f"Successfully launched PowerPoint with {template_file}")
+                else:
+                    log.error(f"Error opening PowerPoint: {result.stderr.strip()}")
+            except Exception as e:
+                log.error(f"Exception opening PowerPoint on Windows: {e}", exc_info=True)
+                
+        elif system_platform == "Darwin":  # macOS
+            log.info(f"Attempting to open {template_file} with PowerPoint on macOS...")
+            cmd = ['open', '-a', 'Microsoft PowerPoint', str(template_file)]
+            result = subprocess.run(cmd, check=False, capture_output=True, text=True)
+            
+            if result.returncode == 0:
+                log.info(f"Successfully opened {template_file} with PowerPoint.")
+            else:
+                log.error(f"Failed to open {template_file} with PowerPoint. Error: {result.stderr.strip()}")
+        else:
+            log.warning(f"Opening PowerPoint on {system_platform} not explicitly supported")
+
+    except Exception as e:
+        log.error(f"An unexpected error occurred during PowerPoint preparation: {e}", exc_info=True) 

computer_use_ootb_internal/preparation/star_rail_prepare.py

@@ -1,100 +1,100 @@
-# src/computer_use_ootb_internal/preparation/star_rail_prepare.py
-import time
-import platform
-import subprocess # Added for taskkill
-import pyautogui
-import webbrowser
-import logging # Use logging instead of print for better practice
-
-# Set up logging for this module if needed, or rely on root logger
-log = logging.getLogger(__name__)
-
-def run_preparation(state):
-    """
-    Performs environment preparation specific to Star Rail on Windows.
-    Closes existing Edge browsers, opens the specified URL in a new Edge instance,
-    and performs initial clicks.
-    """
-    if platform.system() != "Windows":
-        log.info("Star Rail preparation skipped: Not running on Windows.")
-        return
-
-    log.info("Star Rail preparation: Starting environment setup on Windows...")
-    url = "https://sr.mihoyo.com/cloud/#/" # Consider making this configurable later
-    browser_opened = False
-    try:
-        # Attempt to close existing Microsoft Edge processes
-        log.info("Attempting to close existing Microsoft Edge processes...")
-        try:
-            # /F forces termination, /IM specifies image name
-            result = subprocess.run(['taskkill', '/F', '/IM', 'msedge.exe'],
-                                    capture_output=True, text=True, check=False)
-            if result.returncode == 0:
-                log.info("Successfully sent termination signal to msedge.exe processes.")
-            elif "not found" in result.stderr.lower() or "not found" in result.stdout.lower():
-                 log.info("No running msedge.exe processes found to close.")
-            else:
-                 log.warning(f"taskkill command finished with return code {result.returncode}. Output: {result.stdout} Stderr: {result.stderr}")
-            time.sleep(2) # Give processes time to close
-        except FileNotFoundError:
-            log.error("Error: 'taskkill' command not found. Make sure it's in the system PATH.")
-        except Exception as e:
-            log.error(f"Error occurred while trying to close Edge: {e}", exc_info=True)
-
-        # Use only webbrowser.open
-        log.info(f"Attempting to open {url} using webbrowser.open()...")
-        if webbrowser.open(url):
-            log.info(f"Successfully requested browser to open {url} via webbrowser.open().")
-            browser_opened = True
-            # Ensure sleep time for browser load before clicks is present
-            time.sleep(5)
-        else:
-            log.warning("webbrowser.open() returned False, indicating potential failure.")
-
-        if not browser_opened:
-            log.error("Failed to confirm browser opening via webbrowser.open(). Will still attempt clicks.")
-
-        # Add pyautogui click after attempting to open the browser
-        log.info("Proceeding with pyautogui actions...")
-        time.sleep(5) # Wait time for the browser to load
-
-        # Get screen size
-        screen_width, screen_height = pyautogui.size()
-        log.info(f"Detected screen size: {screen_width}x{screen_height}")
-
-        # Calculate click coordinates based on a reference resolution (e.g., 1280x720)
-        # TODO: Make these coordinates more robust or configurable
-        click_x_1 = int(screen_width * (1036 / 1280))
-        click_y_1 = int(screen_height * (500 / 720))
-        log.info(f"Calculated click coordinates for starting the game: ({click_x_1}, {click_y_1})")
-        click_x_2 = int(screen_width * (1233 / 1280))
-        click_y_2 = int(screen_height * (30 / 720))
-        log.info(f"Calculated click coordinates for closing the browser warning: ({click_x_2}, {click_y_2})")
-
-        # Disable failsafe before clicking
-        pyautogui.FAILSAFE = False
-        log.info("PyAutoGUI failsafe temporarily disabled.")
-
-        log.info(f"Clicking at coordinates: ({click_x_1}, {click_y_1})")
-        pyautogui.click(click_x_1, click_y_1)
-        time.sleep(2)
-        pyautogui.click(click_x_1, click_y_1) # Double click?
-
-        # Press F11 to attempt fullscreen
-        log.info("Pressing F11 to enter fullscreen...")
-        time.sleep(1) # Short delay before pressing F11
-        pyautogui.press('f11')
-        time.sleep(1)
-        log.info(f"Clicking at coordinates: ({click_x_2}, {click_y_2})")
-        pyautogui.click(click_x_2, click_y_2)
-        time.sleep(1)
-        pyautogui.click(click_x_2, click_y_2)
-
-        log.info("Star Rail preparation clicks completed.")
-
-    except Exception as e:
-        log.error(f"Error during Star Rail preparation (browser/click): {e}", exc_info=True)
-    finally:
-         # Ensure failsafe is re-enabled
-         pyautogui.FAILSAFE = True
+# src/computer_use_ootb_internal/preparation/star_rail_prepare.py
+import time
+import platform
+import subprocess # Added for taskkill
+import pyautogui
+import webbrowser
+import logging # Use logging instead of print for better practice
+
+# Set up logging for this module if needed, or rely on root logger
+log = logging.getLogger(__name__)
+
+def run_preparation(state):
+    """
+    Performs environment preparation specific to Star Rail on Windows.
+    Closes existing Edge browsers, opens the specified URL in a new Edge instance,
+    and performs initial clicks.
+    """
+    if platform.system() != "Windows":
+        log.info("Star Rail preparation skipped: Not running on Windows.")
+        return
+
+    log.info("Star Rail preparation: Starting environment setup on Windows...")
+    url = "https://sr.mihoyo.com/cloud/#/" # Consider making this configurable later
+    browser_opened = False
+    try:
+        # Attempt to close existing Microsoft Edge processes
+        log.info("Attempting to close existing Microsoft Edge processes...")
+        try:
+            # /F forces termination, /IM specifies image name
+            result = subprocess.run(['taskkill', '/F', '/IM', 'msedge.exe'],
+                                    capture_output=True, text=True, check=False)
+            if result.returncode == 0:
+                log.info("Successfully sent termination signal to msedge.exe processes.")
+            elif "not found" in result.stderr.lower() or "not found" in result.stdout.lower():
+                 log.info("No running msedge.exe processes found to close.")
+            else:
+                 log.warning(f"taskkill command finished with return code {result.returncode}. Output: {result.stdout} Stderr: {result.stderr}")
+            time.sleep(2) # Give processes time to close
+        except FileNotFoundError:
+            log.error("Error: 'taskkill' command not found. Make sure it's in the system PATH.")
+        except Exception as e:
+            log.error(f"Error occurred while trying to close Edge: {e}", exc_info=True)
+
+        # Use only webbrowser.open
+        log.info(f"Attempting to open {url} using webbrowser.open()...")
+        if webbrowser.open(url):
+            log.info(f"Successfully requested browser to open {url} via webbrowser.open().")
+            browser_opened = True
+            # Ensure sleep time for browser load before clicks is present
+            time.sleep(5)
+        else:
+            log.warning("webbrowser.open() returned False, indicating potential failure.")
+
+        if not browser_opened:
+            log.error("Failed to confirm browser opening via webbrowser.open(). Will still attempt clicks.")
+
+        # Add pyautogui click after attempting to open the browser
+        log.info("Proceeding with pyautogui actions...")
+        time.sleep(5) # Wait time for the browser to load
+
+        # Get screen size
+        screen_width, screen_height = pyautogui.size()
+        log.info(f"Detected screen size: {screen_width}x{screen_height}")
+
+        # Calculate click coordinates based on a reference resolution (e.g., 1280x720)
+        # TODO: Make these coordinates more robust or configurable
+        click_x_1 = int(screen_width * (1036 / 1280))
+        click_y_1 = int(screen_height * (500 / 720))
+        log.info(f"Calculated click coordinates for starting the game: ({click_x_1}, {click_y_1})")
+        click_x_2 = int(screen_width * (1233 / 1280))
+        click_y_2 = int(screen_height * (30 / 720))
+        log.info(f"Calculated click coordinates for closing the browser warning: ({click_x_2}, {click_y_2})")
+
+        # Disable failsafe before clicking
+        pyautogui.FAILSAFE = False
+        log.info("PyAutoGUI failsafe temporarily disabled.")
+
+        log.info(f"Clicking at coordinates: ({click_x_1}, {click_y_1})")
+        pyautogui.click(click_x_1, click_y_1)
+        time.sleep(2)
+        pyautogui.click(click_x_1, click_y_1) # Double click?
+
+        # Press F11 to attempt fullscreen
+        log.info("Pressing F11 to enter fullscreen...")
+        time.sleep(1) # Short delay before pressing F11
+        pyautogui.press('f11')
+        time.sleep(1)
+        log.info(f"Clicking at coordinates: ({click_x_2}, {click_y_2})")
+        pyautogui.click(click_x_2, click_y_2)
+        time.sleep(1)
+        pyautogui.click(click_x_2, click_y_2)
+
+        log.info("Star Rail preparation clicks completed.")
+
+    except Exception as e:
+        log.error(f"Error during Star Rail preparation (browser/click): {e}", exc_info=True)
+    finally:
+         # Ensure failsafe is re-enabled
+         pyautogui.FAILSAFE = True
          log.info("PyAutoGUI failsafe re-enabled.")