gpt_computer_agent/top_bar_wrapper.py

import functools
import time


def wrapper(func):
    """A decorator that logs the start and end of the function call."""

    @functools.wraps(func)
    def wrapped_func(*args, **kwargs):
        try:
            from .gpt_computer_agent import the_main_window

            print("GOOGLE-searching")
            function_name = "Tool: " + func.__name__
            the_main_window.active_border_animation(function_name)
            time.sleep(2)
            result = func(*args, **kwargs)
            the_main_window.deactive_border_animation(function_name)
            time.sleep(1)
            print("GOOGLE SEARCHİNG COMPLEATES")

            return result
        except:
            return func(*args, **kwargs)

    return wrapped_func