diff --git a/src/streamlit_app.py b/src/streamlit_app.py
index 4c87624..2ad50ac 100644
--- a/src/streamlit_app.py
+++ b/src/streamlit_app.py
@@ -19,7 +19,7 @@ from main import (
 
 def is_running_locally():
     """Check if the app is running locally or in cloud environment"""
-    return TKINTER_AVAILABLE
+    return os.environ.get('STREAMLIT_SERVER_ADDRESS', '').startswith('localhost')
 
 # Configure page
 st.set_page_config(