Update app.py

app.py

@@ -7,6 +7,7 @@ import time
 import os
 import tempfile
 import random
+import re  # Für reguläre Ausdrücke zur Gewichtserkennung
 
 # === AUTOMATISCHE NEGATIVE PROMPT GENERIERUNG ===
 def auto_negative_prompt(positive_prompt):
@@ -301,7 +302,44 @@ def text_to_image(prompt, steps, guidance_scale, progress=gr.Progress()):
             
         print(f"Starting generation for: {prompt}")
         start_time = time.time()
+
+        # Liste von Qualitätswörtern/Gewichten, die auf Benutzereingaben prüfen
+        quality_keywords = ['masterpiece', 'best quality', 'high quality', 'highly detailed', 
+                           'exquisite', 'detailed', 'ultra detailed', 'professional', 
+                           'perfect', 'excellent', 'amazing', 'stunning', 'beautiful']
+        
+        # Prüfe, ob der Benutzer bereits Qualitätswörter/Gewichte verwendet hat
+        user_has_quality_words = False
+        
+        # Konvertiere Prompt zu Kleinbuchstaben für die Prüfung
+        prompt_lower = prompt.lower()
+        
+        # Prüfe auf einfache Qualitätswörter
+        for keyword in quality_keywords:
+            if keyword in prompt_lower:
+                user_has_quality_words = True
+                print(f"✓ Benutzer verwendet bereits Qualitätswort: {keyword}")
+                break
+        
+        # Prüfe auf Gewichte (z.B. (word:1.5), [word], etc.)
+        weight_patterns = [r'\([^)]+:\d+(\.\d+)?\)', r'\[[^\]]+\]']
+        for pattern in weight_patterns:
+            if re.search(pattern, prompt):
+                user_has_quality_words = True
+                print("✓ Benutzer verwendet bereits Gewichte im Prompt")
+                break
         
+        # Prompt basierend auf Prüfung anpassen
+        if not user_has_quality_words:
+            enhanced_prompt = f"masterpiece, raw, best quality, highly detailed, {prompt}"
+            print(f"🔄 Verbesserter Prompt: {enhanced_prompt}")
+        else:
+            enhanced_prompt = prompt
+            print("✓ Benutzerprompt wird unverändert verwendet")
+        
+        print(f"Finaler Prompt für Generation: {enhanced_prompt}")
+
+             
         progress(0, desc="Generierung läuft - CPU benötigt bis zu 10 Minuten!")
         
         pipe = load_txt2img()
@@ -317,7 +355,7 @@ def text_to_image(prompt, steps, guidance_scale, progress=gr.Progress()):
         callback = TextToImageProgressCallback(progress, steps)
         
         image = pipe(
-            prompt=prompt,
+            prompt=enhanced_prompt,
             negative_prompt=neg_prompt,  # Automatischen Negativ-Prompt verwenden
             height=IMG_SIZE,
             width=IMG_SIZE,