Update app.py

app.py

@@ -28,11 +28,11 @@ def create_face_mask(image, bbox_coords, face_preserve):
             # GESICHTSERHALTUNG: Maske um das Gesicht herum zeichnen
             draw.rectangle([0, 0, image.size[0], image.size[1]], fill=255)  # Alles weiß = verändern
             draw.rectangle([x1, y1, x2, y2], fill=0)  # Gesicht schwarz = geschützt (rechteckig)
-            print("✅ Gesicht wird GESCHÜTZT - Umgebung wird verändert (rechteckige Maske)")
+            print("Gesicht wird GESCHÜTZT - Umgebung wird verändert (rechteckige Maske)")
         else:
             # NUR GESICHT VERÄNDERN: Nur das Gesicht wird weiß (verändert)
             draw.rectangle([x1, y1, x2, y2], fill=255)  # Gesicht weiß = verändern (rechteckig)
-            print("✅ Nur Gesicht wird verändert - Umgebung bleibt erhalten (rechteckige Maske)")
+            print("Nur Gesicht wird verändert - Umgebung bleibt erhalten (rechteckige Maske)")
     
     return mask
 
@@ -49,7 +49,7 @@ def auto_detect_face_area(image):
     # Stelle sicher, dass Koordinaten innerhalb des Bildes liegen
     x1, y1 = max(0, int(x1)), max(0, int(y1))
     x2, y2 = min(width, int(x2)), min(height, int(y2))
-    print(f"✅ Geschätzte Gesichtskoordinaten: [{x1}, {y1}, {x2}, {y2}]")
+    print(f"Geschätzte Gesichtskoordinaten: [{x1}, {y1}, {x2}, {y2}]")
     return [x1, y1, x2, y2]
 
 # === PIPELINES ===
@@ -86,7 +86,7 @@ def load_img2img():
                 safety_checker=None
             ).to(device)
         except Exception as e:
-            print(f"❌ Fehler beim Laden des Modells: {e}")
+            print(f"Fehler beim Laden des Modells: {e}")
             raise
 
         
@@ -104,6 +104,20 @@ def load_img2img():
 
     return pipe_img2img
 
+# === HILFSFUNKTION FÜR ROBUSTE ZWISCHENSPEICHERUNG ===
+def save_with_fallback(image):
+    """Speichert Bild temporär und gibt geladenes Image zurück (für Gradio-Stabilität)"""
+    try:
+        temp_dir = "/tmp/gradio_fallback"
+        os.makedirs(temp_dir, exist_ok=True)
+        fd, path = tempfile.mkstemp(suffix=".png", dir=temp_dir)
+        os.close(fd)
+        image.save(path, "PNG")
+        return Image.open(path)
+    except Exception as e:
+        print(f"Temp-Speicherung fehlgeschlagen: {e}")
+        return image
+
 # === FUNKTIONEN ===
 def text_to_image(prompt, steps, guidance_scale):
     try:
@@ -118,7 +132,7 @@ def text_to_image(prompt, steps, guidance_scale):
         # ZUFÄLLIGER SEED für Variation
         seed = random.randint(0, 2**32 - 1)
         generator = torch.Generator(device=device).manual_seed(seed)
-        print(f"🎲 Using seed: {seed}")
+        print(f"Using seed: {seed}")
         
         image = pipe(
             prompt=prompt,
@@ -130,12 +144,13 @@ def text_to_image(prompt, steps, guidance_scale):
         ).images[0]
         
         end_time = time.time()
-        print(f"✅ Bild generiert in {end_time - start_time:.2f} Sekunden")
+        print(f"Bild generiert in {end_time - start_time:.2f} Sekunden")
         
-        return image
+        # Robuste Zwischenspeicherung
+        return save_with_fallback(image)
         
     except Exception as e:
-        print(f"❌ Fehler: {e}")
+        print(f"Fehler: {e}")
         import traceback
         traceback.print_exc()
         return None
@@ -145,7 +160,7 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale, fac
         if image is None:
             return None
 
-        print(f"🧩 Img2Img Start → Strength: {strength}, Steps: {steps}, Guidance: {guidance_scale}")
+        print(f"Img2Img Start → Strength: {strength}, Steps: {steps}, Guidance: {guidance_scale}")
         print(f"Prompt: {prompt}")
         print(f"Gesicht beibehalten: {face_preserve}")
         start_time = time.time()
@@ -160,7 +175,7 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale, fac
         # ZUFÄLLIGER SEED für Variation
         seed = random.randint(0, 2**32 - 1)
         generator = torch.Generator(device=device).manual_seed(seed)
-        print(f"🎲 Using seed: {seed}")
+        print(f"Using seed: {seed}")
 
         # --- GESICHTSMASKE ---
         mask = None
@@ -179,15 +194,15 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale, fac
                 int(bbox_y2 * scale_y)
             ]
             bbox_coords = scaled_coords
-            print(f"📐 Skalierte Koordinaten: {scaled_coords}")
+            print(f"Skalierte Koordinaten: {scaled_coords}")
 
         # Maskenlogik basierend auf face_preserve
         if bbox_coords:
             mask = create_face_mask(img_resized, bbox_coords, face_preserve)
             if mask:
-                print("✅ Maske erfolgreich erstellt")
+                print("Maske erfolgreich erstellt")
         else:
-            print("⚠️ Keine gültigen Koordinaten - keine Maske angewendet")
+            print("Keine gültigen Koordinaten - keine Maske angewendet")
             mask = None
 
         # --- PIPELINE-AUFRUF ---
@@ -203,25 +218,15 @@ def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale, fac
         )
         
         end_time = time.time()
-        print(f"✅ Bild transformiert in {end_time - start_time:.2f} Sekunden")
+        print(f"Bild transformiert in {end_time - start_time:.2f} Sekunden")
         
         generated_image = result.images[0]
 
-        # Temp-Speicherung
-        try:
-            temp_dir = "/tmp/gradio_fallback"
-            os.makedirs(temp_dir, exist_ok=True)
-            temp_path = os.path.join(temp_dir, f"generated_{int(time.time())}.png")
-            generated_image.save(temp_path, "PNG")
-            saved_image = Image.open(temp_path)
-        except Exception as temp_error:
-            print(f"⚠️ Temp-Speicherung fehlgeschlagen: {temp_error}")
-            saved_image = generated_image
-        
-        return saved_image
+        # Robuste Zwischenspeicherung
+        return save_with_fallback(generated_image)
         
     except Exception as e:
-        print(f"❌ Fehler: {e}")
+        print(f"Fehler: {e}")
         import traceback
         traceback.print_exc()
         return None
@@ -270,7 +275,6 @@ def main_ui():
         gr.Markdown(
             """
             # Demo-Projekt: Stable Diffusion Text-to-Image / Image-to-Image
-
             <br>
             
             <div class="info-box">
@@ -287,7 +291,6 @@ def main_ui():
             die die <br><strong>gesamte Komplexität einer professionellen Umsetzung</strong> verdeutlicht.<br><br>
             Die damit gezeigten Fähigkeiten meinerseits sind <strong>sicherlich auf andere Projekte übertragbar</strong>.
             </div>
-
             <br><br>
             
             <div class="info-box">
@@ -299,7 +302,6 @@ def main_ui():
             generierte Bild wird in diesem Fall <strong>nicht angezeigt</strong>. Dies gewährleistet die Stabilität des Systems und verhindert Ressourcenkonflikte. <br>
             Die Meldung "Connection re-established" zeigt an, dass die Verbindung wiederhergestellt wurde, die laufende Verarbeitung jedoch priorisiert wird.
             </div>
-
             <br>
             
             <div class="info-box">
@@ -308,7 +310,6 @@ def main_ui():
             &nbsp;&nbsp;• die gezielte <strong>Veränderung des Gesichts</strong> bei erhaltener Umgebung, oder<br>
             &nbsp;&nbsp;• den <strong>Schutz des Motivs</strong>, während nur der <strong>Hintergrund angepasst</strong> wird.
             </div>
-
             <br><br>
             
             <div class="info-box">
@@ -360,7 +361,7 @@ def main_ui():
                             info="Wie stark der Prompt befolgt wird (7-12 für gute Balance)"
                         )
                 
-                generate_btn = gr.Button("🎨 Bild generieren", variant="primary")
+                generate_btn = gr.Button("Bild generieren", variant="primary")
                 txt_output = gr.Image(
                     label="Generiertes Bild", 
                     show_download_button=True
@@ -422,13 +423,13 @@ def main_ui():
                 # GESICHTSOPTIONEN
                 with gr.Row():
                     face_preserve = gr.Checkbox(
-                        label="👤 Gesicht beibehalten", 
+                        label="Gesicht beibehalten", 
                         value=True,
                         info="Aktiviert: Gesicht bleibt erhalten, Hintergrund wird verändert | Deaktiviert: Nur Gesicht wird verändert"
                     )
                 
                 with gr.Row():
-                    gr.Markdown("**🔍 Gesichtsbereich anpassen**")
+                    gr.Markdown("**Gesichtsbereich anpassen**")
                 
                 with gr.Row():
                     bbox_x1 = gr.Number(
@@ -467,7 +468,7 @@ def main_ui():
                         )
                 
                 
-                transform_btn = gr.Button("🔄 Bild transformieren", variant="primary")
+                transform_btn = gr.Button("Bild transformieren", variant="primary")
                 
                 with gr.Row():
                     img_output = gr.Image(