Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - OPTIMIERTE VERSION
+# app.py - KORRIGIERTE VERSION
 import gradio as gr
 from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
 import torch
@@ -8,8 +8,8 @@ import time
 # === OPTIMIERTE EINSTELLUNGEN ===
 device = "cuda" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if device == "cuda" else torch.float32
-IMG_SIZE = 512 #320  # Bessere Qualität aber noch schnell
-STEPS = 35  #20      # Mehr Schritte für bessere Qualität
+IMG_SIZE = 512
+STEPS = 35
 
 print(f"Running on: {device}")
 
@@ -56,13 +56,12 @@ def text_to_image(prompt):
         
         pipe = load_txt2img()
         
-        # Optimierte Generierung für bessere Qualität
         image = pipe(
             prompt=prompt,
             height=IMG_SIZE,
             width=IMG_SIZE,
             num_inference_steps=STEPS,
-            guidance_scale=7.5,  # Standard für bessere Qualität
+            guidance_scale=7.5,
         ).images[0]
         
         end_time = time.time()
@@ -76,13 +75,14 @@ def text_to_image(prompt):
         traceback.print_exc()
         return None
 
-# IMAGE-TO-IMAGE FUNKTION FIXEN:
-def img_to_image(image, prompt="", strength=0.6):  # Strength Range anpassen
+def img_to_image(image, prompt="", strength=0.6):
     try:
         if image is None:
             return None
             
         print(f"Image-to-Image mit Strength: {strength}")
+        start_time = time.time()
+        
         pipe = load_img2img()
         img_resized = image.convert("RGB").resize((512, 512))
         
@@ -94,20 +94,6 @@ def img_to_image(image, prompt="", strength=0.6):  # Strength Range anpassen
             guidance_scale=7.5
         )
         
-        print("✅ Bild 100% generiert - sende an UI")
-        return result.images[0]
-        
-    except Exception as e:
-        print(f"❌ Fehler: {e}")
-        return None
-
-# SLIDER ANPASSEN:
-strength_slider = gr.Slider(
-    0.5, 0.8, 0.6,  # Jetzt ab 0.5 möglich!
-    label="Stärke der Veränderung"
-)
-
-
         end_time = time.time()
         print(f"✅ Bild transformiert in {end_time - start_time:.2f} Sekunden")
         
@@ -155,23 +141,23 @@ with gr.Blocks() as app:
         
         with gr.Row():
             img_prompt = gr.Textbox(
-                placeholder="z.B. convert to winter landscape with snow",
+                placeholder="background only: winter forest, keep girl and snowman unchanged",
                 lines=2,
                 label="Transformations-Prompt (Englisch)"
             )
         
         with gr.Row():
             strength_slider = gr.Slider(
-                0.6, 0.9, 0.75, 
+                0.5, 0.8, 0.6,  # Jetzt ab 0.5 möglich!
                 label="Stärke der Veränderung"
             )
         
         with gr.Row():
             gr.Markdown(
                 "**Stärke-Einstellung:**  "
-                "• **0.6-0.7:** Behält mehr vom Original bei  "
-                "• **0.75:** Gute Balance (empfohlen)  "  
-                "• **0.8-0.9:** Stärkere Veränderungen"
+                "• **0.5-0.6:** Behält fast alles Original bei  "
+                "• **0.65-0.7:** Gute Balance  "  
+                "• **0.75-0.8:** Stärkere Veränderungen"
             )
         
         transform_btn = gr.Button("🔄 Bild transformieren", variant="primary")