fix

safety/auth/utils.py

@@ -1,6 +1,7 @@
 from functools import lru_cache
 import json
 import logging
+import sys
 from typing import Any, Optional, Dict, Callable, Tuple
 from authlib.integrations.requests_client import OAuth2Session
 from authlib.integrations.base_client.errors import OAuthError
@@ -430,24 +431,14 @@ def send(self, request: requests.PreparedRequest, **kwargs: Any) -> requests.Res
 
 @lru_cache(maxsize=1)
 def is_jupyter_notebook() -> bool:
-    """
-    Check if the code is being executed inside a Jupyter Notebook.
-
-    Returns:
-        bool: True if running in a Jupyter notebook, False otherwise.
-    """
     try:
         from IPython import get_ipython
         shell = get_ipython()
-        if shell:
-            shell_name = shell.__class__.__name__
-            if shell_name == "ZMQInteractiveShell":
-                print("Running in a Jupyter Notebook!")
-            elif shell_name == "TerminalInteractiveShell":
-                print("Running in an IPython terminal!")
-            else:
-                print("Running in a different IPython environment!")
-        else:
-            print("Not running in an IPython environment.")
-    except ImportError:
-        print("IPython is not installed. Running in a standard Python environment.")
+        if shell and 'IPKernelApp' in shell.config:
+            # Detect Jupyter and Google Colab environments
+            return True
+        elif "google.colab" in sys.modules:
+            return True
+    except (ImportError, AttributeError):
+        return False
+    return False