Add final demo check for Spaces

app.py

@@ -926,8 +926,18 @@ else:
     # Local execution: demo will be created in main()
     demo = None
 
+# Final check: ensure demo is set for Spaces
+if IS_SPACES and demo is None:
+    logger.error("CRITICAL: Demo is None on Spaces! Creating fallback demo.")
+    try:
+        with gr.Blocks() as fallback_demo:
+            gr.Markdown("# Error: Demo was not created properly\n\nPlease check the logs for details.")
+        demo = fallback_demo
+    except:
+        pass
+
 # For local execution only (not on Spaces)
 if __name__ == "__main__":
     # Don't run main() on Spaces - demo is already created at module level
-    if not (os.getenv("SPACE_ID") or os.getenv("SYSTEM") == "spaces"):
+    if not IS_SPACES:
         main()