Update app.py

app.py

@@ -37,6 +37,41 @@ style_list = [
         "prompt": "hyper-realistic 4K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
         "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
     },
+    {
+        "name": "HDR",
+        "prompt": "HDR photo of {prompt}. high dynamic range, vivid colors, sharp contrast, realistic, detailed, high resolution, professional",
+        "negative_prompt": "dull, low contrast, blurry, unrealistic, cartoonish, ugly, deformed",
+    },
+    {
+        "name": "Cinematic",
+        "prompt": "cinematic still {prompt}. emotional, harmonious, vignette, highly detailed, high budget, bokeh, cinemascope, moody, epic, gorgeous, film grain, grainy",
+        "negative_prompt": "anime, cartoon, graphic, text, painting, crayon, graphite, abstract, glitch, deformed, mutated, ugly, disfigured",
+    },
+    {
+        "name": "Photo",
+        "prompt": "cinematic photo {prompt}. 35mm photograph, film, bokeh, professional, 4k, highly detailed",
+        "negative_prompt": "drawing, painting, crayon, sketch, graphite, impressionist, noisy, blurry, soft, deformed, ugly",
+    },
+    {
+        "name": "Anime",
+        "prompt": "anime artwork {prompt}. anime style, key visual, vibrant, studio anime, highly detailed",
+        "negative_prompt": "photo, deformed, black and white, realism, disfigured, low contrast",
+    },
+    {
+        "name": "Manga",
+        "prompt": "manga style {prompt}. vibrant, high-energy, detailed, iconic, Japanese comic style",
+        "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, Western comic style",
+    },
+    {
+        "name": "Digital",
+        "prompt": "concept art {prompt}. digital artwork, illustrative, painterly, matte painting, highly detailed",
+        "negative_prompt": "photo, photorealistic, realism, ugly",
+    },
+    {
+        "name": "3D Model",
+        "prompt": "professional 3d model {prompt}. octane render, highly detailed, volumetric, dramatic lighting",
+        "negative_prompt": "ugly, deformed, noisy, low poly, blurry, painting",
+    },
     {
         "name": "(No style)",
         "prompt": "{prompt}",
@@ -176,17 +211,32 @@ with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
                 placeholder="Enter your prompt",
                 container=False,
             )
-            run_button = gr.Button("Run", scale=0)
-        result = gr.Gallery(label="Result", columns=NUM_IMAGES_PER_PROMPT, show_label=False)
+            run_button = gr.Button("Run")
+        result = gr.Gallery(label="Result", columns=1, preview=True)
     with gr.Accordion("Advanced options", open=False):
+        use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True, visible=True)
+        negative_prompt = gr.Text(
+            label="Negative prompt",
+            max_lines=1,
+            placeholder="Enter a negative prompt",
+            value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, NSFW",
+            visible=True,
+        )
         with gr.Row():
-            use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True, visible=True)
-            negative_prompt = gr.Text(
-                label="Negative prompt",
-                max_lines=1,
-                placeholder="Enter a negative prompt",
-                value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, NSFW",
-                visible=True,
+            num_inference_steps = gr.Slider(
+                label="Steps",
+                minimum=10,
+                maximum=60,
+                step=1,
+                value=30,
+            )
+        with gr.Row():
+            num_images_per_prompt = gr.Slider(
+                label="Images",
+                minimum=1,
+                maximum=5,
+                step=1,
+                value=2,
             )
         seed = gr.Slider(
             label="Seed",
@@ -194,33 +244,41 @@ with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
             maximum=MAX_SEED,
             step=1,
             value=0,
+            visible=True
         )
         randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-        style = gr.Dropdown(label="Style", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
         with gr.Row(visible=True):
             width = gr.Slider(
                 label="Width",
-                minimum=256,
-                maximum=MAX_IMAGE_SIZE,
-                step=32,
+                minimum=512,
+                maximum=2048,
+                step=8,
                 value=1024,
             )
             height = gr.Slider(
                 label="Height",
-                minimum=256,
-                maximum=MAX_IMAGE_SIZE,
-                step=32,
+                minimum=512,
+                maximum=2048,
+                step=8,
                 value=1024,
             )
         with gr.Row():
             guidance_scale = gr.Slider(
                 label="Guidance Scale",
                 minimum=0.1,
-                maximum=20,
+                maximum=20.0,
                 step=0.1,
-                value=3.0,
+                value=6,
             )
-
+    with gr.Row(visible=True):
+        style_selection = gr.Radio(
+            show_label=True,
+            container=True,
+            interactive=True,
+            choices=STYLE_NAMES,
+            value=DEFAULT_STYLE_NAME,
+            label="Image Style",
+        )
     gr.Examples(
         examples=examples,
         inputs=prompt,
@@ -247,7 +305,7 @@ with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
             prompt,
             negative_prompt,
             use_negative_prompt,
-            style,
+            style_selection,
             seed,
             width,
             height,