Update app.py

app.py

@@ -6,19 +6,19 @@ from style_template import styles
 import base64
 
 prodia = Prodia()
-STYLE_NAMES = list(styles.keys())
-DEFAULT_STYLE_NAME = "Photographic (Default)"
+STYLE_PRESETS = [None, "3d-model", "analog-film", "anime", "cinematic", "comic-book", "digital-art", "enhance", "fantasy-art", "isometric", "line-art", "low-poly",
+                 "neon-punk", "origami", "photographic", "pixel-art", "texture", "craft-clay"]
 MAX_SEED = np.iinfo(np.int32).max
 
 
-def generate_image(upload_images, prompt, negative_prompt, style_name, steps, strength, seed, progress=gr.Progress(track_tqdm=True)):
+def generate_image(upload_images, prompt, negative_prompt, style_preset, steps, strength, seed, progress=gr.Progress(track_tqdm=True)):
     error_if_no_img(prompt)
-    p, n = apply_style(style_name, prompt, negative_prompt)
 
     job = prodia.create("/photomaker",
                         imageData=[file_to_base64(img) for img in upload_images],
-                        prompt=p,
-                        negative_prompt=n,
+                        prompt=prompt,
+                        negative_prompt=negative_prompt,
+                        style_preset=style_preset,
                         steps=steps,
                         strength=strength,
                         seed=seed if seed != 0 else None
@@ -45,11 +45,6 @@ def remove_back_to_files():
     return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
 
 
-def apply_style(style_name: str, positive: str, negative: str = "") -> tuple[str, str]:
-    p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
-    return p.replace("{prompt}", positive), n + ' ' + negative
-
-
 def get_image_path_list(folder_name):
     image_basename_list = os.listdir(folder_name)
     image_path_list = sorted([os.path.join(folder_name, basename) for basename in image_basename_list])
@@ -68,13 +63,13 @@ def get_example():
         [
             get_image_path_list('./examples/scarletthead_woman'),
             "instagram photo, portrait photo of a woman img, colorful, perfect face, natural skin, hard shadows, film grain",
-            "(No style)",
+            None,
             "(asymmetry, worst quality, low quality, illustration, 3d, 2d, painting, cartoons, sketch), open mouth",
         ],
         [
             get_image_path_list('./examples/newton_man'),
             "sci-fi, closeup portrait photo of a man img wearing the sunglasses in Iron man suit, face, slim body, high quality, film grain",
-            "(No style)",
+            None,
             "(asymmetry, worst quality, low quality, illustration, 3d, 2d, painting, cartoons, sketch), open mouth",
         ],
     ]
@@ -104,7 +99,7 @@ with gr.Blocks(css=css) as demo:
             prompt = gr.Textbox(label="Prompt",
                                 info="Try something like 'a photo of a man/woman img', 'img' is the trigger word.",
                                 placeholder="A photo of a [man/woman img]...")
-            style = gr.Dropdown(label="Style template", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
+            style = gr.Dropdown(label="Style template", choices=STYLE_PRESETS, value=None)
             submit = gr.Button("Submit")
 
             with gr.Accordion(open=False, label="Advanced Options"):