Update app.py

app.py

@@ -1,18 +1,75 @@
 import gradio as gr
+import subprocess
+import os
 import spaces
 
-@spaces.GPU  # Nutzt ZeroGPU, wenn aktiviert
-def on_button_click():
-    return "Button wurde geklickt!"
+# Funktion zum Installieren von Infinigen (einmalig beim Start)
+def install_infinigen():
+    infinigen_dir = "infinigen"
+    if not os.path.exists(infinigen_dir):
+        print("Klone Infinigen...")
+        subprocess.run(["git", "clone", "https://github.com/princeton-vl/infinigen.git"], check=True)
+        
+        # Blender-Python-Pfad (anpassen falls nötig)
+        blender_python = "/usr/share/blender/4.2/python/bin/python3.10"
+        if not os.path.exists(blender_python):
+            # Fallback: Suche nach Blender-Python
+            blender_python = subprocess.run(
+                ["find", "/usr", "-name", "python3.10", "-path", "*/blender/*"],
+                capture_output=True, text=True
+            ).stdout.strip().split("\n")[0]
+            print(f"Blender Python gefunden unter: {blender_python}")
+        
+        # Installiere Infinigen mit Blender-Python
+        subprocess.run([
+            blender_python, "-m", "pip", "install", "-e", f"{infinigen_dir}[terrain,vis]", "--no-deps"
+        ], check=True)
+        print("Infinigen installiert.")
 
-with gr.Blocks() as demo:
-    button = gr.Button("Klick mich")
-    output = gr.Textbox(label="Ausgabe")
+# Infinigen beim Start installieren
+install_infinigen()
+
+# Funktion zum Generieren einer Szene mit ZeroGPU
+@spaces.GPU
+def generate_scene(seed):
+    output_dir = "outputs"
+    os.makedirs(output_dir, exist_ok=True)
+    
+    blender_python = "/usr/share/blender/4.2/python/bin/python3.10"  # Muss ggf. angepasst werden
+    command = [
+        blender_python, "-m", "infinigen.datagen.manage_jobs",
+        "--output_folder", output_dir,
+        "--num_scenes", "1",
+        "--specific_seed", str(int(seed)),
+        "--configs", "infinigen/infinigen_examples/configs/desert.gin",
+        "infinigen/infinigen_examples/configs/simple.gin",
+        "--pipeline_configs", "infinigen/infinigen_examples/configs/local_16GB.gin",
+        "infinigen/infinigen_examples/configs/monocular.gin",
+        "infinigen/infinigen_examples/configs/blender_gt.gin"
+    ]
+    
+    try:
+        result = subprocess.run(command, capture_output=True, text=True, check=True)
+        print("STDOUT:", result.stdout)
+        print("STDERR:", result.stderr)
+        output_path = os.path.join(output_dir, "0000000000.png")
+        if os.path.exists(output_path):
+            return output_path
+        return f"Fehler: Bild nicht gefunden. STDERR: {result.stderr}"
+    except subprocess.CalledProcessError as e:
+        return f"Fehler: {e.stderr}"
+
+# Gradio-Oberfläche
+with gr.Blocks(title="Infinigen Demo") as demo:
+    gr.Markdown("## Infinigen Scene Generator")
+    seed_input = gr.Number(label="Seed", value=0, precision=0)
+    output_image = gr.Image(label="Generierte Szene")
+    generate_button = gr.Button("Szene generieren")
     
-    button.click(
-        fn=on_button_click,
-        inputs=None,
-        outputs=output
+    generate_button.click(
+        fn=generate_scene,
+        inputs=[seed_input],
+        outputs=[output_image]
     )
 
 demo.launch()  # Kein share=True