Update app.py

app.py

@@ -30,7 +30,15 @@ def start_session(req: gr.Request):
     
 def end_session(req: gr.Request):
     user_dir = os.path.join(CUSTOM_SAVE_DIR, str(req.session_hash))
-    shutil.rmtree(user_dir)
+    try:
+        if os.path.exists(user_dir):
+            shutil.rmtree(user_dir)
+            print(f"[Session Cleanup] Deleted {user_dir}")
+        else:
+            print(f"[Session Cleanup] Directory already gone: {user_dir}")
+    except Exception as e:
+        print(f"[Session Cleanup Warning] {e}")
+
 
 
 def preprocess_image(image: Image.Image) -> Image.Image: