Update app.py

app.py

@@ -11,43 +11,52 @@ class InfinigenManager:
         self.output_dir = Path(output_dir)
         self.blender_bin = None
         self.blender_python = None
+        self.initialized = False
         self._setup_blender()
         self._setup_infinigen()
 
     def _setup_blender(self):
-        """Stellt sicher, dass die neueste Blender-Version installiert ist."""
+        """Stellt sicher, dass Blender installiert ist und findet den Python-Interpreter."""
         try:
-            # Prüfe vorhandene Blender-Version
             blender_bins = glob.glob("/usr/bin/blender")
-            if blender_bins:
-                self.blender_bin = blender_bins[0]
-                version_result = subprocess.run([self.blender_bin, "-v"], capture_output=True, text=True)
-                blender_version = next((line.split()[1] for line in version_result.stdout.split("\n") if "Blender" in line and "." in line), None)
-                print(f"Vorhandene Blender-Version: {blender_version or 'unbekannt'}")
-            else:
-                raise FileNotFoundError("Blender nicht vorhanden")
+            if not blender_bins:
+                raise FileNotFoundError("Blender-Binär nicht gefunden!")
             
-            # Finde Blender-Python
-            base_path = "/usr/share/blender"
+            self.blender_bin = blender_bins[0]
+            version_result = subprocess.run([self.blender_bin, "-v"], capture_output=True, text=True)
+            blender_version = next((line.split()[1] for line in version_result.stdout.split("\n") if "Blender" in line and "." in line), None)
+            print(f"Vorhandene Blender-Version: {blender_version or 'unbekannt'}")
+            
+            # Breite Suche nach Python-Interpreter
+            base_paths = ["/usr/share/blender", "/usr/lib/blender"]
             python_patterns = [
-                f"{base_path}/{blender_version}/python/bin/python*",
-                f"{base_path}/*/python/bin/python*",
-                "/usr/lib/blender/*/python/bin/python*"
+                f"/usr/share/blender/{blender_version}/python/bin/python*",
+                f"/usr/share/blender/*/python/bin/python*",
+                f"/usr/lib/blender/*/python/bin/python*",
+                "/**/blender/*/python/bin/python*"
             ]
             for pattern in python_patterns:
                 python_bins = glob.glob(pattern, recursive=True)
                 if python_bins:
                     self.blender_python = python_bins[0]
                     print(f"Blender Python gefunden: {self.blender_python}")
+                    self.initialized = True
                     return
             
+            # Debugging: Vollständige Python-Suche
+            all_python = glob.glob("/**/python", recursive=True)
+            print(f"Alle Python-Pfade: {all_python}")
             raise RuntimeError("Blender Python-Interpreter nicht gefunden!")
         except Exception as e:
             print(f"Fehler bei Blender-Setup: {e}")
-            raise
+            self.blender_python = None
 
     def _setup_infinigen(self):
-        """Installiert Infinigen, falls nicht vorhanden."""
+        """Installiert Infinigen, falls möglich."""
+        if not self.blender_python:
+            print("Kann Infinigen nicht installieren: Blender Python fehlt.")
+            return
+        
         if not self.base_dir.exists():
             print("Klone Infinigen...")
             subprocess.run(["git", "clone", "https://github.com/princeton-vl/infinigen.git"], check=True)
@@ -61,6 +70,8 @@ class InfinigenManager:
 
     def _is_infinigen_installed(self):
         """Prüft, ob Infinigen bereits installiert ist."""
+        if not self.blender_python:
+            return False
         try:
             subprocess.run([self.blender_python, "-c", "import infinigen"], check=True, capture_output=True)
             return True
@@ -70,8 +81,10 @@ class InfinigenManager:
     @spaces.GPU
     def generate_scene(self, seed, configs=None, pipeline_configs=None):
         """Generiert eine Szene mit Infinigen."""
-        self.output_dir.mkdir(exist_ok=True)
+        if not self.initialized or not self.blender_python:
+            return "Fehler: Infinigen nicht initialisiert (Blender Python fehlt)!"
         
+        self.output_dir.mkdir(exist_ok=True)
         configs = configs or ["infinigen_examples/configs/desert.gin", "infinigen_examples/configs/simple.gin"]
         pipeline_configs = pipeline_configs or [
             "infinigen_examples/configs/local_16GB.gin",
@@ -96,10 +109,7 @@ class InfinigenManager:
             return f"Fehler: {e.stderr}"
 
 # Manager initialisieren
-try:
-    manager = InfinigenManager()
-except Exception as e:
-    print(f"Fehler bei der Initialisierung: {e}")
+manager = InfinigenManager()  # Immer definieren, auch bei Fehlern
 
 # Gradio-Oberfläche
 with gr.Blocks(title="Infinigen Demo") as demo: