Update app.py

app.py

@@ -17,7 +17,7 @@ IMG_SIZE = 512
 
 print(f"Running on: {device}")
 
-# === MODELLKONFIGURATION ===
+# === MODELLKONFIGURATION (NUR 2 MODELLE) ===
 MODEL_CONFIGS = {
     "runwayml/stable-diffusion-v1-5": {
         "name": "🏠 Stable Diffusion 1.5 (Universal)",
@@ -25,7 +25,7 @@ MODEL_CONFIGS = {
         "requires_vae": False,
         "recommended_steps": 35,
         "recommended_cfg": 7.5,
-        "supports_fp16": True  # Offizielles Modell hat FP16 Variante
+        "supports_fp16": True
     },
     "SG161222/Realistic_Vision_V6.0_B1_noVAE": {
         "name": "👤 Realistic Vision V6.0 (Portraits)",
@@ -34,40 +34,12 @@ MODEL_CONFIGS = {
         "vae_model": "stabilityai/sd-vae-ft-mse",
         "recommended_steps": 40,
         "recommended_cfg": 7.0,
-        "supports_fp16": False  # Custom Model, keine FP16 Variante
-    },
-    "RunDiffusion/Juggernaut-X-v10": {
-        "name": "🏢 Juggernaut X (Business)",
-        "description": "Ideal for corporate images, team photos, professional settings",
-        "requires_vae": False,
-        "recommended_steps": 35,
-        "recommended_cfg": 7.5,
-        "supports_fp16": False  # Custom Model, keine FP16 Variante
-    },
-    "Lykon/DreamShaper": {
-        "name": "🎨 DreamShaper (Artistic)",
-        "description": "Creative interpretations, artistic styles, illustrations",
-        "requires_vae": False,
-        "recommended_steps": 40,
-        "recommended_cfg": 8.0,
-        "supports_fp16": False  # Custom Model, keine FP16 Variante
-    },
-    "nitrosocke/redshift-diffusion": {
-        "name": "🖼️ Redshift Diffusion (Design)",
-        "description": "Clean CGI style, product visuals, design mockups",
-        "requires_vae": False,
-        "recommended_steps": 30,
-        "recommended_cfg": 8.5,
-        "supports_fp16": False  # Custom Model, keine FP16 Variante
+        "supports_fp16": False
     }
 }
 
-
 # === SAFETENSORS KONFIGURATION ===
-SAFETENSORS_MODELS = [
-    "runwayml/stable-diffusion-v1-5",
-    "stabilityai/stable-diffusion-2-1"
-]
+SAFETENSORS_MODELS = ["runwayml/stable-diffusion-v1-5"]
 
 # Aktuell ausgewähltes Modell (wird vom User gesetzt)
 current_model_id = "runwayml/stable-diffusion-v1-5"
@@ -79,11 +51,19 @@ def auto_negative_prompt(positive_prompt):
     negatives = []
 
     # Personen / Portraits
-    if any(w in p for w in ["person", "man", "woman", "face", "portrait", "team", "employee", "people", "crowd"]):
+    if any(w in p for w in [
+    "person", "man", "woman", "face", "portrait", "team", "employee", 
+    "people", "crowd", "character", "figure", "human", "child", "baby",
+    "girl", "boy", "lady", "gentleman", "fairy", "elf", "dwarf", "orc",
+    "mermaid", "angel", "demon", "witch", "wizard", "creature", "being",
+    "model", "actor", "actress", "celebrity", "avatar", "persona"]):
         negatives.append(
-            "bad anatomy, malformed hands, extra fingers, uneven eyes, distorted face, unrealistic skin, mutated, deformed, ugly, disfigured, poorly drawn face, missing limbs, extra limbs, fused fingers, too many fingers, long neck"
+            "bad anatomy, malformed hands, extra fingers, uneven eyes, distorted face, "
+            "unrealistic skin, mutated, deformed, ugly, disfigured, poorly drawn face, "
+            "missing limbs, extra limbs, fused fingers, too many fingers, bad teeth, "
+            "mutated hands, long neck, extra wings, multiple wings"
         )
-
+  
     # Business / Corporate
     if any(w in p for w in ["office", "business", "team", "meeting", "corporate", "company", "workplace"]):
         negatives.append(
@@ -194,7 +174,6 @@ def load_txt2img(model_id):
                 print(f"⚠️ Fehler beim Laden der VAE: {vae_error}")
                 print("ℹ️ Versuche ohne VAE weiter...")
                 vae = None
-
         
         # Modellparameter basierend auf Modelltyp
         model_params = {
@@ -202,21 +181,18 @@ def load_txt2img(model_id):
             "safety_checker": None,
             "requires_safety_checker": False,
             "add_watermarker": False,
-            "allow_pickle": True,  # 1. Immer True für .bin
-        }        
+            "allow_pickle": True,  # Für .bin Modelle wichtig
+        }
         
-              
-
-        # SAFETENSORS LOGIK HIER EINFÜGEN:
+        # SAFETENSORS LOGIK
         if model_id in SAFETENSORS_MODELS:
             model_params["use_safetensors"] = True
             print(f"ℹ️ Verwende safetensors für {model_id}")
         else:
             model_params["use_safetensors"] = False
             print(f"ℹ️ Verwende .bin weights für {model_id}")
-     
         
-        # NUR FP16 Variante laden wenn Modell sie unterstützt UND wir auf GPU sind
+        # FP16 Variante nur wenn Modell sie unterstützt UND wir auf GPU sind
         if config.get("supports_fp16", False) and torch_dtype == torch.float16:
             model_params["variant"] = "fp16"
             print("ℹ️ Verwende FP16 Variante")
@@ -272,7 +248,6 @@ def load_txt2img(model_id):
         except Exception as scheduler_error:
             print(f"⚠️ Konnte DPM-Scheduler nicht setzen: {scheduler_error}")
             print("ℹ️ Verwende Standard-Scheduler weiter")
-            # Behalte den aktuellen Scheduler bei
         
         # Optimierungen
         pipe_txt2img.enable_attention_slicing()
@@ -727,7 +702,7 @@ def main_ui():
                 
                 with gr.Row():
                     with gr.Column(scale=2):
-                        # Modellauswahl Dropdown
+                        # Modellauswahl Dropdown (NUR 2 MODELLE)
                         model_dropdown = gr.Dropdown(
                             choices=[
                                 (config["name"], model_id) 
@@ -735,7 +710,7 @@ def main_ui():
                             ],
                             value="runwayml/stable-diffusion-v1-5",
                             label="📁 Modellauswahl",
-                            info="Wähle ein Modell basierend auf deinem Anwendungsfall"
+                            info="🏠 Universal vs 👤 Portraits"
                         )
                         
                         # Modellinformationen Box
@@ -966,5 +941,5 @@ if __name__ == "__main__":
         max_file_size="10MB",
         show_error=True,
         share=False,
-        ssr_mode=False  # ← DAS FEHLT!
+        ssr_mode=False  # SSR deaktivieren für Stabilität
     )