Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-from gradio import TempFile
 from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
 from diffusers import StableDiffusionInpaintPipeline
 import torch
@@ -105,21 +104,6 @@ def load_img2img():
 
     return pipe_img2img
 
-# === HILFSFUNKTION FÜR ROBUSTE ZWISCHENSPEICHERUNG ===
-temp_file_manager = TempFile()
-
-def save_with_fallback(image):
-    try:
-        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):
@@ -150,7 +134,7 @@ def text_to_image(prompt, steps, guidance_scale):
         print(f"Bild generiert in {end_time - start_time:.2f} Sekunden")
         
         # Robuste Zwischenspeicherung
-        return save_with_fallback(image)
+        return image
         
     except Exception as e:
         print(f"Fehler: {e}")
@@ -225,8 +209,7 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale, fac
         
         generated_image = result.images[0]
 
-        # Robuste Zwischenspeicherung
-        return save_with_fallback(generated_image)
+        return generated_image
         
     except Exception as e:
         print(f"Fehler: {e}")
@@ -368,7 +351,7 @@ def main_ui():
                 txt_output = gr.Image(
                     label="Generiertes Bild", 
                     show_download_button=True,
-                    type="filepath"
+                    type="pil"
                 )
                 
                 generate_btn.click(
@@ -478,7 +461,7 @@ def main_ui():
                     img_output = gr.Image(
                         label="Transformiertes Bild",
                         show_download_button=True,
-                        type="filepath"
+                        type="pil"
                     )
                 
                 # Event-Handler für Bild-Upload