Update ActivityMonitor.py

fixed the shit opus missed

Author: tilltmk

ActivityMonitor.py

@@ -1,14 +1,24 @@
+import time
+import os
+import platform
+import subprocess
+import threading
 import customtkinter as ctk
 import keyboard
 import clipboard
-import pygetwindow as gw
-import pyscreenshot as ImageGrab
-import time
-import markdown2
-import threading
-import os
+import re
 import ollama
 
+if platform.system() == "Linux":
+    import mss
+    import mss.tools
+elif platform.system() == "Windows":
+    import pyscreenshot as ImageGrab
+    import pygetwindow as gw
+
+screenshot_directory = "screenshots"
+archive_directory = "archives"
+
 def generate_answers(model_name, prompt):
     model_name = "vicuna:13b-16k"
     messages = [
@@ -23,6 +33,62 @@ def generate_answers(model_name, prompt):
     except Exception as e:
         print(f"There was an error communicating with the Ollama model: {e}")
         return None
+    
+def get_active_window_title():
+    """
+    Get the title of the active window.
+    
+    Returns:
+        str: The title of the active window.
+    """
+    if platform.system() == "Linux":
+        try:
+            # Get the window ID of the currently focused window
+            window_id = subprocess.check_output(["xdotool", "getactivewindow"]).decode().strip()
+
+            # Get the name of the window with this ID
+            window_name = subprocess.check_output(["xdotool", "getwindowname", window_id]).decode().strip()
+
+            # Get the geometry of the window (position and size)
+            geom_output = subprocess.check_output(["xdotool", "getwindowgeometry", "--shell", window_id]).decode()
+
+            # Use a dictionary to store variables from the shell output
+            geom_vars = dict(re.findall(r'(\w+)=(\S+)', geom_output))
+
+            # Convert values to integers
+            x = int(geom_vars['X'])
+            y = int(geom_vars['Y'])
+            width = int(geom_vars['WIDTH'])
+            height = int(geom_vars['HEIGHT'])
+
+            with mss.mss() as sct:
+                # The screenshot region is a tuple: (x, y, width, height)
+                monitor = {"top": y, "left": x, "width": width, "height": height}
+                screenshot_path = os.path.join(screenshot_directory, f"{time.time()}_screenshot.png")
+                # Capture the specified region
+                sct_img = sct.grab(monitor)
+                mss.tools.to_png(sct_img.rgb, sct_img.size, output=screenshot_path)
+            
+            return window_name
+        except subprocess.CalledProcessError as e:
+            print(f"Error retrieving window title: {e}")
+            return None
+    elif platform.system() == "Windows":
+        # Capture active window and screenshot
+        active_window = gw.getActiveWindow()
+        if active_window:
+            try:
+                # save screenshot
+                screenshot_path = os.path.join(screenshot_directory, f"{time.time()}_screenshot.png")
+                screenshot = ImageGrab.grab(bbox=active_window.box)
+                screenshot.save(screenshot_path)
+
+                # save title and screenshot in markdown log
+                window_title = active_window.title if active_window.title else "Unknown Window"
+                return window_title
+            except Exception as e:
+                print(f"Error getting active window: {e}")
+                return None
 
 class ActivityMonitor(ctk.CTk):
     def __init__(self):
@@ -70,8 +136,8 @@ def stop_monitoring(self):
         self.status_label.configure(text="Status: Stopped")
 
     def analyze_with_ollama(self):
-        prompt = f"Please analyze the following activities and add your comments:\n\n{self.markdown_log}"
-        detailed_description = generate_answers("vicuna:13b-16k", prompt)
+        prompt: str = f"Please analyze the following activities and add your comments:\n\n{self.markdown_log}"
+        detailed_description = generate_answers(model_name="vicuna:13b-16k", prompt=prompt)
         if detailed_description:
             self.markdown_log += f"\n\n## Ollama's Analysis\n\n{detailed_description}"
             self.write_markdown_file()
@@ -84,8 +150,9 @@ def quit_monitoring(self):
     def monitor_activities(self):
         previous_clipboard = ""
         previous_window = None
-        screenshot_directory = "screenshots"
         os.makedirs(screenshot_directory, exist_ok=True)
+        os.makedirs(archive_directory, exist_ok=True)
+        window_title = None
 
         while self.is_running:
             # Capture clipboard content
@@ -95,20 +162,10 @@ def monitor_activities(self):
                 previous_clipboard = current_clipboard
 
             # Capture active window and screenshot
-            active_window = gw.getActiveWindow()
-            if active_window and (active_window != previous_window):
-                try:
-                    # save screenshot
-                    screenshot_path = os.path.join(screenshot_directory, f"{time.time()}_screenshot.png")
-                    screenshot = ImageGrab.grab(bbox=active_window.box)
-                    screenshot.save(screenshot_path)
-
-                    # save title and screenshot in markdown log
-                    window_title = active_window.title if active_window.title else "Unknown Window"
-                    self.markdown_log += f"\n## Window Selected: {window_title}\n\n![Screenshot]({screenshot_path})\n"
-                    previous_window = active_window
-                except Exception as e:
-                    print(f"Error getting active window: {e}")
+            window_title = get_active_window_title()
+            if previous_window != window_title:
+                self.markdown_log += f"\n## Window Selected: {window_title}\n\n![Screenshot]({screenshot_directory}/{time.time()}_screenshot.png)\n"
+                previous_window = window_title
 
             # Capture typed keys (improved version)
             key_events = keyboard.record(until='enter')
@@ -117,8 +174,37 @@ def monitor_activities(self):
                 self.markdown_log += f"\n- **Typed Text**: {typed_text.replace('space', ' ')}\n"
                 self.text_buffer = ""
 
+            # Archive URLs
+            self.archive_url(archive_directory)
+
             time.sleep(1)  # Reduce CPU usage
 
+    def archive_url(self, directory):
+        """
+        Archive URLs.
+        
+        Args:
+            directory (str): The directory to save the archived URLs.
+        """
+        window_title = get_active_window_title()
+        # Extend or adjust the list of browser window titles as needed
+        browsers = ["Brave", "Firefox", "Chrome", "Chromium", "Edge", "Opera"]
+        if any(browser in window_title for browser in browsers):
+            # Set focus to the address bar and copy URL
+            keyboard.send('ctrl+l')  # Focus on address bar
+            time.sleep(0.1)  # Briefly wait for focus to be set
+            keyboard.send('ctrl+c')  # Copy URL
+            time.sleep(0.1)  # Wait for clipboard to update
+            keyboard.send('esc')  # Escape
+
+            # Read URL from clipboard
+            url = clipboard.paste()
+            if url.startswith("http") or url.startswith("https"):
+                with open(os.path.join(directory, f"{time.time()}_url.txt"), "w", encoding='utf-8') as file:
+                    file.write(url)
+        else:
+            print("The active window is not a recognized browser window.")
+
     def write_markdown_file(self):
         with open("activity_log.md", "w", encoding='utf-8') as md_file:
             md_file.write(self.markdown_log)