feat: Implement logging functionality; add logger configuration and retrieval

ai/compute.py

@@ -52,7 +52,6 @@ def reprompt(nextsteps: str, processor) -> None:
     scr = screenshot_to_base64(take_screenshot())
     return processor.process(nextsteps, img_data=scr)
 
-
 def _execute(name, args, processor):
     if name == "click_button":
         press_mouse(MouseInput(**args))

ai/processor.py

@@ -4,6 +4,10 @@ import openai
 from flask import jsonify
 from objects import aic
 import ai.compute
+from objects import logger as logger_module
+import logging
+
+logger: logging.Logger = logger_module.get_logger(__name__)
 
 class AIProcessor:
     def __init__(self, api_key: str, model: str = "gpt-4.1"):
@@ -62,6 +66,12 @@ class AIProcessor:
                         processor=self,
                     )
                     outputs.append(r) if r else None
+                # Make sure the two last messages from user has an image, but set disable_image to True for the others
+                for msg in self.session.messages[-2:]:
+                    msg.disable_image = True
+                logger.debug(
+                    self.session.messages
+                )
                 if reexec:
                     img = ai.compute.screenshot_to_base64(
                         ai.compute.take_screenshot()
@@ -86,15 +96,6 @@ class AIProcessor:
                 aic.Message(role="assistant", content=output_text)
             )
 
-            # re-execute if needed
-            if reexec:
-                img = ai.compute.screenshot_to_base64(
-                    ai.compute.take_screenshot()
-                )
-
-                outputs.append(
-                    *self.process(nextsteps, img)
-                )
             return outputs
         except Exception as e:
             traceback.print_exc()