Update app.py

app.py

@@ -159,20 +159,24 @@ except Exception as e:
 CODEFORMER_PATH = "CodeFormer/inference_codeformer.py"
 
 def ensure_codeformer():
+    """
+    Ensure CodeFormer's local basicsr + facelib are importable and
+    pretrained weights are downloaded. No setup.py needed — we use
+    sys.path / PYTHONPATH instead.
+    """
     try:
         if not os.path.exists("CodeFormer"):
             logger.info("CodeFormer not found, cloning repository...")
             subprocess.run("git clone https://github.com/sczhou/CodeFormer.git", shell=True, check=True)
             subprocess.run("pip install -r CodeFormer/requirements.txt", shell=True, check=False)
         
-        # FIX: Must cd into CodeFormer directory so setup.py can find ./basicsr/VERSION
-        if os.path.exists("CodeFormer/basicsr/setup.py"):
-            logger.info("Installing BasicSR from local directory...")
-            subprocess.run(
-                "cd CodeFormer && python basicsr/setup.py develop",
-                shell=True, check=True, timeout=120
-            )
-            logger.info("BasicSR installed successfully")
+        # Add CodeFormer root to sys.path so `import basicsr` and
+        # `import facelib` resolve to the local (compatible) versions
+        # instead of the broken PyPI basicsr==1.4.2.
+        codeformer_root = os.path.join(os.getcwd(), "CodeFormer")
+        if codeformer_root not in sys.path:
+            sys.path.insert(0, codeformer_root)
+            logger.info(f"Added {codeformer_root} to sys.path for local basicsr/facelib")
         
         try:
             import realesrgan
@@ -180,8 +184,11 @@ def ensure_codeformer():
         except ImportError:
             logger.info("Installing RealESRGAN...")
             subprocess.run("pip install --no-cache-dir realesrgan", shell=True, check=True, timeout=120)
+            # Uninstall the broken PyPI basicsr that realesrgan pulls in
+            subprocess.run("pip uninstall -y basicsr", shell=True, check=False, timeout=30)
             logger.info("RealESRGAN installed successfully")
         
+        # Download pretrained weights if not already present
         if os.path.exists("CodeFormer"):
             try:
                 subprocess.run("python CodeFormer/scripts/download_pretrained_models.py facelib", shell=True, check=False, timeout=300)
@@ -1992,6 +1999,7 @@ if __name__ == "__main__":
 
 
 
+
 # #####################FASTAPI___________________##############
 # import os
 # os.environ["OMP_NUM_THREADS"] = "1"