Update app.py

app.py

@@ -41,27 +41,33 @@ def patch_file_content(filename):
     try:
         with open(filename, "r", encoding="utf-8") as f: content = f.read()
         
-        # 1. GPU -> CPU
+        # GPU -> CPU
         content = content.replace("device = 'cuda'", "device = 'cpu'")
         content = content.replace('device = "cuda"', 'device = "cpu"')
         content = content.replace("torch.device('cuda')", "torch.device('cpu')")
         content = content.replace(".to(device)", "")
         content = content.replace(".cuda()", ".cpu()")
         
-        # 2. Disable Driver Checks
+        # Disable Driver Checks
         content = re.sub(r'.*Found no NVIDIA driver.*', '    pass', content)
         content = re.sub(r'.*torch.cuda.is_available().*', '    pass', content)
         content = re.sub(r'.*torch._C._cuda_init().*', '    pass', content)
         
-        # 3. Fix State Errors
+        # Fix State Errors
         old_func = "def on_select_mask_tab(state):"
         new_func = "def on_select_mask_tab(state):" + NL + "    if not state or 'sample' not in state: return None"
         content = content.replace(old_func, new_func)
         
-        # 4. Fix NameError: inverse_image (مهم برای آپلود عکس)
-        if "from draggan.web import" not in content and "def on_image_change" in content:
-             # اضافه کردن ایمپورت دستی اگر وجود ندارد
-             content = "from draggan.stylegan2.inversion import inverse_image" + NL + content
+        # Fix NameError: inverse_image (CORRECTED PATH)
+        # The file is likely at draggan/stylegan2/inversion.py, but usually exposed via draggan.web or similiar
+        # We will inject the import directly from the file path to be safe
+        if "def on_image_change" in content and "inverse_image" not in content:
+             # Try importing from the known structure
+             import_line = "from draggan.stylegan2.inversion import inverse_image" + NL
+             # If the previous attempt failed, it might be because 'draggan' package isn't fully resolvable this way
+             # Let's try a relative import fix or just rely on the fact we add cwd to sys.path later
+             if "from draggan.web import" not in content:
+                 content = import_line + content
 
         with open(filename, "w", encoding="utf-8") as f: f.write(content)
         print("Patched: " + filename)
@@ -70,11 +76,9 @@ def patch_file_content(filename):
 
 def recursive_patch():
     print("Starting deep patch...")
-    # فایل‌های اصلی
     patch_file_content("gradio_app.py")
     patch_file_content("visualizer_drag_gradio.py")
     
-    # تمام فایل‌های داخلی
     for root, dirs, files in os.walk("draggan"):
         for file in files:
             if file.endswith(".py"):
@@ -123,7 +127,13 @@ def load_app(filename):
         current_dir = os.getcwd()
         if current_dir not in sys.path: sys.path.append(current_dir)
         
-        # Force CPU globally
+        # Manually import the missing module to ensure it exists before app loads
+        try:
+            import draggan.stylegan2.inversion
+            print("Successfully pre-imported inversion module.")
+        except ImportError:
+            print("Could not pre-import inversion module. Path might be wrong.")
+
         torch.cuda.is_available = lambda: False
         torch.backends.cudnn.enabled = False