From 888b082eb6e6c8e7daf51751c14781320a29ea9d Mon Sep 17 00:00:00 2001
From: Eric Lay <ericlaytm@gmail.com>
Date: Wed, 27 Mar 2024 16:18:03 -0500
Subject: [PATCH] Update app.py

Update model escape shell page to load after template
---
 app.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/app.py b/app.py
index 780fc84..149b384 100644
--- a/app.py
+++ b/app.py
@@ -22,14 +22,17 @@ def current_website_dir():
     if request_count % 5 == 0:
         global website_dir
         website_dir = str(ROOT_DIR+"/Marvel")
+    elif request_count % 6 == 0:
+        website_dir = str(ROOT_DIR+"/Escape")
     else:
         global WEBSITE_DIRS
         try:
+            WEBSITE_DIRS.remove('Escape')
             WEBSITE_DIRS.remove('Marvel')
-        except ValueError:
+        except (ValueError, IndexError) as e:
             pass
         website_dir = random.choice(WEBSITE_DIRS)
-        
+
 # Make static files available
 @app.route('/<path:filename>', methods=['GET'])
 def static_proxy(filename):