diff --git a/backend/backup_restore_api.py b/backend/backup_restore_api.py index f22bcea..12e14ff 100644 --- a/backend/backup_restore_api.py +++ b/backend/backup_restore_api.py @@ -16,10 +16,23 @@ load_dotenv() def get_environment() -> str: - """Detect environment based on FRONTEND_URL""" - frontend_url = os.getenv('FRONTEND_URL', 'http://localhost:5174') + """Detect environment based on FRONTEND_URL or deployment environment""" + # Check FRONTEND_URL first + frontend_url = os.getenv('FRONTEND_URL', '') if 'myrecipes.dvirlabs.com' in frontend_url or 'my-recipes.dvirlabs.com' in frontend_url: return 'prod' + + # Check if explicitly set via ENV variable + env_var = os.getenv('ENVIRONMENT', '').lower() + if env_var in ['prod', 'production']: + return 'prod' + + # Check database host (if not localhost, likely production) + db_host = os.getenv('DB_HOST', 'localhost') + if db_host not in ['localhost', '127.0.0.1', 'postgres']: + return 'prod' + + # Default to dev return 'dev'