✨ add colab checker

Signed-off-by: peter szemraj <[email protected]>

app.py

@@ -238,4 +238,6 @@ with gr.Blocks(theme=theme, analytics_enabled=False, css=_styles) as demo:
 demo.queue(max_size=10).launch(
     debug=True,
     server_port=DEFAULT_PORT,
+    show_api=False,
+    share=utils.is_google_colab(),
 )

utils.py

@@ -4,6 +4,16 @@ from typing import List
 from settings import STATIC_PATH
 
 
+def is_google_colab():
+    """Check if the environment is Google Colab."""
+    try:
+        from google.colab import drive
+
+        return True
+    except ImportError:
+        return False
+
+
 def get_file_as_string(file_name, path=STATIC_PATH) -> str:
     """Loads the content of a file given its name
     and returns all of its lines as a single string