Update app.py

app.py

@@ -105,25 +105,23 @@ def load_img2img():
     return pipe_img2img
 
 # === HILFSFUNKTION FÜR ROBUSTE ZWISCHENSPEICHERUNG ===
+from gradio import TempFile
+
+temp_file_manager = TempFile()
+
 def save_with_fallback(image):
     try:
-        temp_dir = os.getenv("GRADIO_TEMP_DIR") or tempfile.gettempdir()
-        temp_file = tempfile.NamedTemporaryFile(
-            suffix=".png", 
-            dir=temp_dir, 
-            delete=False   # ← Datei bleibt im HF-Container!
-        )
-        path = temp_file.name
-        temp_file.close()
-        
-        image.save(path, "PNG")
-        print(f"HF-Temp gespeichert: {path}")
-        return path
+        with temp_file_manager as temp_file:
+            path = temp_file.name
+            image.save(path, "PNG")
+            print(f"Gradio-Temp gespeichert: {path}")
+            return path
     except Exception as e:
         print(f"Temp-Speicherung fehlgeschlagen: {e}")
         return image
 
 
+
 # === FUNKTIONEN ===
 def text_to_image(prompt, steps, guidance_scale):
     try: