Update app.py

app.py

@@ -3,6 +3,8 @@ from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
 import torch
 from PIL import Image
 import time
+import os
+import tempfile
 
 # === OPTIMIERTE EINSTELLUNGEN ===
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -85,7 +87,7 @@ def text_to_image(prompt, steps, guidance_scale):
         end_time = time.time()
         print(f"✅ Bild generiert in {end_time - start_time:.2f} Sekunden")
         
-        return image
+        return image  # Direkte Rückgabe aus RAM
         
     except Exception as e:
         print(f"❌ Fehler: {e}")
@@ -121,7 +123,27 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale):
         end_time = time.time()
         print(f"✅ Bild transformiert in {end_time - start_time:.2f} Sekunden")
         
-        return result.images[0]
+        generated_image = result.images[0]  # Bild aus RAM
+        
+        # === TEMP-SPEICHERUNG ALS FALLBACK ===
+        try:
+            # Erstelle Temp-Verzeichnis falls nicht vorhanden
+            temp_dir = "/tmp/gradio_fallback"
+            os.makedirs(temp_dir, exist_ok=True)
+            
+            # Speichere Bild temporär
+            temp_path = os.path.join(temp_dir, f"generated_{int(time.time())}.png")
+            generated_image.save(temp_path, "PNG")
+            print(f"💾 Bild temporär gespeichert: {temp_path}")
+            
+            # Lade Bild wieder für konsistente Rückgabe
+            saved_image = Image.open(temp_path)
+            
+        except Exception as temp_error:
+            print(f"⚠️ Temp-Speicherung fehlgeschlagen: {temp_error}")
+            saved_image = generated_image  # Fallback auf RAM-Bild
+        
+        return saved_image  # Rückgabe des gespeicherten oder RAM-Bildes
         
     except Exception as e:
         print(f"❌ Fehler: {e}")
@@ -247,6 +269,4 @@ app.launch(
     server_port=7860,
     max_file_size="10MB",
     show_error=True
-    # enable_queue ist bereits standardmäßig True
-    # prevent_thread_lock wird nicht benötigt
 )