Refactor import order and enhance environment detection for spaces; streamline local execution support

app.py

@@ -1,25 +1,10 @@
 import os
-import zipfile
-import tempfile
-import shutil
-import uuid
-from pathlib import Path
-
-import gradio as gr
-import torch
-import numpy as np
-from PIL import Image
-
-from diffusers import Flux2Pipeline
-from diffusers.utils import load_image
-
-import py7zr
-from skimage.metrics import peak_signal_noise_ratio, structural_similarity
-from skimage.util import img_as_float
+import sys
 
 
 # =========================
-# Environment Detection & GPU Decorator Setup
+# CRITICAL: Import spaces FIRST before any CUDA-related packages
+# ZeroGPU requires spaces to be imported before torch/diffusers
 # =========================
 def _is_hf_spaces_env() -> bool:
     """Detect if running in Hugging Face Spaces environment"""
@@ -30,16 +15,15 @@ def _is_hf_spaces_env() -> bool:
 
 IS_HF_SPACES = _is_hf_spaces_env()
 
-# CRITICAL: ZeroGPU scanner requires seeing `import spaces` at module level
-# We use try/except to handle local environment where spaces is not installed
 try:
     import spaces
 
     HAS_SPACES = True
 except ImportError:
     HAS_SPACES = False
-
     # Create dummy module for local environment
+    from types import ModuleType
+
     class _SpacesDummy:
         @staticmethod
         def GPU(duration: int = 180, **kwargs):
@@ -50,13 +34,31 @@ except ImportError:
 
             return decorator
 
-    import sys
-    from types import ModuleType
-
     spaces = ModuleType("spaces")
     spaces.GPU = _SpacesDummy.GPU
     sys.modules["spaces"] = spaces
 
+# =========================
+# Now import CUDA-related packages
+# =========================
+import zipfile
+import tempfile
+import shutil
+import uuid
+from pathlib import Path
+
+import gradio as gr
+import torch
+import numpy as np
+from PIL import Image
+
+from diffusers import Flux2Pipeline
+from diffusers.utils import load_image
+
+import py7zr
+from skimage.metrics import peak_signal_noise_ratio, structural_similarity
+from skimage.util import img_as_float
+
 
 # =========================
 # Config