update templates

Signed-off-by: AnyISalIn <anyisalin@gmail.com>

app.py

@@ -150,7 +150,8 @@ def create_ui():
                         # trained_loras_models = [_.name for _ in get_noviata_client(novita_key).models_v3(refresh=True).filter_by_type("lora").filter_by_visibility("private")]
                         serving_models = [_.models[0].model_name for _ in get_noviata_client(novita_key).list_training().filter_by_model_status("SERVING")]
                         serving_models_labels = [_.task_name for _ in get_noviata_client(novita_key).list_training().filter_by_model_status("SERVING")]
-                        return gr.update(choices=serving_models_labels), gr.update(value=serving_models)
+                        default_serving_model = serving_models_labels[0] if len(serving_models_labels) > 0 else None
+                        return gr.update(choices=serving_models_labels, value=default_serving_model), gr.update(value=serving_models)
 
                     inference_refresh_button.click(
                         inputs=[novita_key],
@@ -158,70 +159,76 @@ def create_ui():
                         fn=inference_refresh_button_fn
                     )
 
-                    templates = [
-                        {
-                            "style_prompt": "(masterpiece), (extremely intricate:1.3), (realistic), portrait of a person, the most handsome in the world, (medieval armor), metal reflections, upper body, outdoors, intense sunlight, far away castle, professional photograph of a stunning person detailed, sharp focus, dramatic, award winning, cinematic lighting, octane render unreal engine, volumetrics dtx, (film grain, blurry background, blurry foreground, bokeh, depth of field, sunset, motion blur:1.3), chainmail",
-                            "style_negative_prompt": "BadDream_53202, UnrealisticDream_53204",
-                            "style_model": "dreamshaper_8_93211.safetensors",
-                            "style_method": "txt2img",
-                            "style_height": 768,
-                            "style_width": 512,
-                            "style_reference_image": "./00001.jpg",
-                        },
-                        # {
-                        #     "style_prompt": "upper body, ((masterpiece)), 1990s style , Student, ID photo, Vintage, Retro, School, Nostalgia",
-                        #     "style_negative_prompt": "BadDream, UnrealisticDream",
-                        #     "style_model": "checkpoint/dreamshaper_8",
-                        #     "style_lora_model": "lora/junmoxiao.safetensors",
-                        #     "style_lora_scale": 1.0,
-                        #     "style_method": "img2img",
-                        #     "style_embeddings": [
-                        #         "embedding/BadDream.pt",
-                        #         "embedding/UnrealisticDream.pt"
-                        #     ],
-                        #     "style_reference_image": "examples/style-2.png",
-                        # }
-                    ]
-
                     first_stage_request_body = gr.JSON(label="First Stage Request Body, POST /api/v2/txt2img")
                     second_stage_request_body = gr.JSON(label="Second Stage Request Body, POST /api/v2/adetailer")
 
-                    def mirror(*args):
-                        return args
+            templates = [
+                {
+                    "style_prompt": "(masterpiece), (extremely intricate:1.3), (realistic), portrait of a person, the most handsome in the world, (medieval armor), metal reflections, upper body, outdoors, intense sunlight, far away castle, professional photograph of a stunning person detailed, sharp focus, dramatic, award winning, cinematic lighting, octane render unreal engine, volumetrics dtx, (film grain, blurry background, blurry foreground, bokeh, depth of field, sunset, motion blur:1.3), chainmail",
+                    "style_negative_prompt": "BadDream_53202, UnrealisticDream_53204",
+                    "style_model": "dreamshaper_8_93211.safetensors",
+                    "style_method": "txt2img",
+                    "style_height": 768,
+                    "style_width": 512,
+                    "style_reference_image": "./00001.jpg",
+                },
+                {
+                    "style_prompt": "photo of beautiful age 18 girl, pastel hair, freckles sexy, beautiful, close up, young, dslr, 8k, 4k, ultrarealistic, realistic, natural skin, textured skin",
+                    "style_negative_prompt": "BadDream_53202, UnrealisticDream_53204",
+                    "style_model": "dreamshaper_8_93211.safetensors",
+                    "style_method": "controlnet-depth",
+                    "style_height": 768,
+                    "style_width": 512,
+                    "style_reference_image": "./00002.jpg",
+                },
+                {
+                    "style_prompt": "majesty, holy, saintly, godly, 1girl, an angel descending from heaven, upper body, beautiful asian goddess, looking at viewer, detail face and eyes, symmetrical eyes, glowing white eye, warm attitude, long hair, blonde hair, floating hair, royal clothes, gold armor, feathered wings, glowing wings, nice hands",
+                    "style_negative_prompt": "BadDream_53202, UnrealisticDream_53204",
+                    "style_model": "dreamshaper_8_93211.safetensors",
+                    "style_method": "controlnet-canny",
+                    "style_height": 768,
+                    "style_width": 512,
+                    "style_reference_image": "./00003.jpg",
+                }
+            ]
+
+            def mirror(*args):
+                return args
 
-                    examples = gr.Examples(
-                        [
-                            [
-                                _.get("style_prompt", ""),
-                                _.get("style_negative_prompt", ""),
-                                _.get("style_model", ""),
-                                _.get("style_height", 512),
-                                _.get("style_width", 512),
-                                _.get("style_method", "txt2img"),
-                                _.get("style_reference_image", ""),
-                            ] for _ in templates
-                        ],
-                        [
-                            style_prompt,
-                            style_negative_prompt,
-                            style_model,
-                            style_height,
-                            style_width,
-                            style_method,
-                            style_reference_image,
-                        ],
+            with gr.Row():
+                examples = gr.Examples(
+                    [
                         [
-                            style_prompt,
-                            style_negative_prompt,
-                            style_model,
-                            style_height,
-                            style_width,
-                            style_method,
-                            style_reference_image,
-                        ],
-                        mirror,
-                        cache_examples=False,
-                    )
+                            _.get("style_prompt", ""),
+                            _.get("style_negative_prompt", ""),
+                            _.get("style_model", ""),
+                            _.get("style_height", 512),
+                            _.get("style_width", 512),
+                            _.get("style_method", "txt2img"),
+                            _.get("style_reference_image", ""),
+                        ] for _ in templates
+                    ],
+                    [
+                        style_prompt,
+                        style_negative_prompt,
+                        style_model,
+                        style_height,
+                        style_width,
+                        style_method,
+                        style_reference_image,
+                    ],
+                    [
+                        style_prompt,
+                        style_negative_prompt,
+                        style_model,
+                        style_height,
+                        style_width,
+                        style_method,
+                        style_reference_image,
+                    ],
+                    mirror,
+                    cache_examples=False,
+                )
 
             def generate(novita_key, gender, style_prompt, style_negative_prompt, style_model, style_lora, _hide_lora_training_response, style_hegiht, style_width, style_method, style_reference_image, num_images):
 
@@ -356,7 +363,8 @@ def create_ui():
             except Exception as e:
                 logging.error(e)
                 return novita_key, gr.update(choices=[], value=None), gr.update(value=None), f"$ UNKNOWN"
-            return novita_key, gr.update(choices=serving_models_labels), gr.update(value=serving_models), f"$ {user_info_json.credit_balance / 100 / 100:.2f}"
+            default_serving_model = serving_models_labels[0] if len(serving_models_labels) > 0 else None
+            return novita_key, gr.update(choices=serving_models_labels, value=default_serving_model), gr.update(value=serving_models), f"$ {user_info_json.credit_balance / 100 / 100:.2f}"
 
         novita_key.change(onload, inputs=novita_key, outputs=[novita_key, style_lora, _hide_lora_training_response, user_balance], _js="(v)=>{ setStorage('novita_key',v); return [v]; }")
 
@@ -369,12 +377,6 @@ def create_ui():
 
     return demo
 
-    # style_method.change(
-    #     inputs=[style_method],
-    #     outputs=[style_reference_image],
-    #     fn=lambda method: gr.update(visible=method in ["controlnet", "img2img", "ip-adapater"])
-    # )
-
 
 if __name__ == '__main__':
     demo = create_ui()