Update app.py

app.py

@@ -65,12 +65,16 @@ def text_to_image(prompt, steps, guidance_scale):
         
         pipe = load_txt2img()
         
+        # Seed für Reproduzierbarkeit (-1 = zufällig)
+        generator = torch.Generator(device=device).manual_seed(-1)
+        
         image = pipe(
             prompt=prompt,
             height=IMG_SIZE,
             width=IMG_SIZE,
             num_inference_steps=steps,
             guidance_scale=guidance_scale,
+            generator=generator
         ).images[0]
         
         end_time = time.time()
@@ -84,23 +88,29 @@ def text_to_image(prompt, steps, guidance_scale):
         traceback.print_exc()
         return None
 
-def img_to_image(image, prompt, strength, steps, guidance_scale):
+def img_to_image(image, prompt, neg_prompt, strength, steps, guidance_scale):
     try:
         if image is None:
             return None
             
         print(f"Image-to-Image mit Strength: {strength}, Steps: {steps}, Guidance: {guidance_scale}")
+        print(f"Negative Prompt: {neg_prompt}")
         start_time = time.time()
         
         pipe = load_img2img()
         img_resized = image.convert("RGB").resize((IMG_SIZE, IMG_SIZE))
         
+        # Seed für Reproduzierbarkeit (-1 = zufällig)
+        generator = torch.Generator(device=device).manual_seed(-1)
+        
         result = pipe(
             prompt=prompt,
+            negative_prompt=neg_prompt,
             image=img_resized,
             strength=strength,
             num_inference_steps=steps,
-            guidance_scale=guidance_scale
+            guidance_scale=guidance_scale,
+            generator=generator
         )
         
         end_time = time.time()
@@ -137,14 +147,14 @@ with gr.Blocks() as app:
             with gr.Column():
                 txt_guidance = gr.Slider(
                     minimum=1.0, maximum=20.0, value=7.5, step=0.5,
-                    label="Guidance Scale (Prompt-Treue)"
+                    label="Guidance Scale (Prompt-Treue vs. Verzerrung)"
                 )
         
         with gr.Row():
             gr.Markdown(
                 "**Parameter-Erklärung:**  "
                 "• **Steps:** Mehr = bessere Qualität, aber langsamer  "
-                "• **Guidance:** Höher = stärkere Prompt-Treue, kann verzerren  "
+                "• **Guidance:** Niedrig = weniger Verzerrung, Hoch = mehr KI-Fantasie  "
             )
         
         generate_btn = gr.Button("🎨 Bild generieren", variant="primary")
@@ -171,11 +181,18 @@ with gr.Blocks() as app:
             )
         
         with gr.Row():
-            img_prompt = gr.Textbox(
-                placeholder="background only: winter forest, keep girl and snowman unchanged",
-                lines=2,
-                label="Transformations-Prompt (Englisch)"
-            )
+            with gr.Column():
+                img_prompt = gr.Textbox(
+                    placeholder="background only: winter forest, keep girl and snowman unchanged",
+                    lines=2,
+                    label="Transformations-Prompt (Englisch)"
+                )
+            with gr.Column():
+                img_neg_prompt = gr.Textbox(
+                    placeholder="blurry, deformed, ugly, bad anatomy, distorted faces",
+                    lines=2,
+                    label="Negativ-Prompt (was vermieden werden soll)"
+                )
         
         with gr.Row():
             with gr.Column():
@@ -191,7 +208,7 @@ with gr.Blocks() as app:
             with gr.Column():
                 img_guidance = gr.Slider(
                     minimum=1.0, maximum=20.0, value=7.5, step=0.5,
-                    label="Guidance Scale (Prompt-Treue)"
+                    label="Guidance Scale (Prompt-Treue vs. Verzerrung)"
                 )
         
         with gr.Row():
@@ -199,7 +216,8 @@ with gr.Blocks() as app:
                 "**Parameter-Erklärung:**  "
                 "• **Strength:** Niedrig = behält Original, Hoch = starke Veränderung  "
                 "• **Steps:** Mehr = bessere Qualität, aber langsamer  "
-                "• **Guidance:** Höher = stärkere Prompt-Treue  "
+                "• **Guidance:** Niedrig = weniger Verzerrung, Hoch = mehr KI-Fantasie  "
+                "• **Negativ-Prompt:** Beschreibt was NICHT im Bild sein soll  "
             )
         
         transform_btn = gr.Button("🔄 Bild transformieren", variant="primary")
@@ -212,7 +230,7 @@ with gr.Blocks() as app:
         
         transform_btn.click(
             fn=img_to_image,
-            inputs=[img_input, img_prompt, strength_slider, img_steps, img_guidance],
+            inputs=[img_input, img_prompt, img_neg_prompt, strength_slider, img_steps, img_guidance],
             outputs=img_output,
             concurrency_limit=1
         )