Update app.py

app.py

@@ -6,6 +6,8 @@ import torch
 from diffusers import DiffusionPipeline, AutoencoderTiny
 from huggingface_hub import hf_hub_download
 
+import config
+
 def feifeimodload():
 
     dtype = torch.bfloat16
@@ -36,7 +38,11 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
-    #prompt = f"{prompt}, Master of Light and Shadow."
+    if styles_Radio:
+        style_name = styles_Radio
+        for style in config.style_list:
+            if style["name"] == style_name:
+                prompt = style["prompt"].replace("{prompt}", prompt)
     image = pipe(
             prompt = "flux, 8k, ",
             prompt_2 = prompt,  
@@ -85,7 +91,13 @@ DarkIdol-flux is a text-to-image AI model designed to create aesthetic, detailed
                 container=False,
                 value="",
             )
-            
+            styles_name = [style["name"] for style in config.style_list]
+            styles_Radio = gr.Dropdown(
+                styles_name,
+                label="Styles",
+                multiselect=False,
+                value="(None)",
+            )
         run_button = gr.Button("Run")
         
         result = gr.Image(label="Result", show_label=False,height=520)