Update app.py

app.py

@@ -9,7 +9,10 @@ from diffusers import (
     AutoencoderTiny,
 )
 from huggingface_hub import hf_hub_download
-
+import config
+styles_name = [style["name"] for style in config.style_list]
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 2048
 
 def feifeimodload():
 
@@ -36,11 +39,9 @@ def feifeimodload():
     return pipe
     
 pipe = feifeimodload()
-MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 2048
 
 @spaces.GPU()
-def infer(prompt="", Expansion_select=False, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, guidancescale=3.5, num_feifei=0.35, nsfw_select=False, nsfw_slider=1, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt="",  styles_Radio="(None)", Expansion_select=False, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, guidancescale=3.5, num_feifei=0.35, nsfw_select=False, nsfw_slider=1, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
@@ -73,7 +74,11 @@ def infer(prompt="", Expansion_select=False, seed=42, randomize_seed=False, widt
         prompt = re.sub(r"young woman", replacement, prompt, flags=re.IGNORECASE)
         prompt = re.sub(r"woman", replacement, prompt, flags=re.IGNORECASE)
         prompt = re.sub(r"model", replacement, prompt, flags=re.IGNORECASE)
-        
+    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,
@@ -109,15 +114,22 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""# FLUX.1 + feifei-flux-lora """)
-        prompt = gr.Text(
-            label="Prompt",
-            show_label=False,
-            max_lines=12,
-            placeholder="Enter your prompt",
-            value="",
-            container=False,
-        )
-   
+        with gr.Row():
+            prompt = gr.Text(
+                label="Prompt",
+                show_label=False,
+                max_lines=12,
+                placeholder="Enter your prompt",
+                value="",
+                container=False,
+            )
+        with gr.Row():
+            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)
         Expansion_select = gr.Checkbox(label="FeiFei Expansion") 
@@ -200,7 +212,7 @@ with gr.Blocks(css=css) as demo:
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn = infer,
-        inputs = [prompt, Expansion_select, seed, randomize_seed, width, height, num_inference_steps, guidancescale, num_feifei, nsfw_select, nsfw_slider],
+        inputs = [prompt, styles_Radio, Expansion_select, seed, randomize_seed, width, height, num_inference_steps, guidancescale, num_feifei, nsfw_select, nsfw_slider],
         outputs = [result, seed]
     )