add free trial

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

app.py

@@ -49,10 +49,12 @@ def create_ui():
             <img src="https://raw.githubusercontent.com/wiki/novitalabs/sd-webui-cleaner/images/logo2.png" width="120px;" alt="Unsplash" />
         </a>
         <h1>Face Stylization Playground</h1>
-        <h3>Powered by <a href="https://novita.ai/?utm_source=huggingface&utm_medium=face-stylization&utm_campaign=face-stylization">Novita.AI</a></h2>
-        <h3> Get Novita.AI API Key from <a href="https://novita.ai/?utm_source=huggingface&utm_medium=face-stylization&utm_campaign=face-stylization">Novita.AI</a></h2>
+        <h3>Start integrate with <a href="https://novita.ai/get-started/Model_training.html?utm_source=huggingface&utm_medium=face-stylization&utm_campaign=face-stylization">Model_training API</a> 
+        <h3> Get Novita.AI API Key from <a href="https://novita.ai/get-started/Account_account_and_key.html?utm_source=huggingface&utm_medium=face-stylization&utm_campaign=face-stylization">Novita.AI</a></h2>
         '''
                 )
+
+        free_trial_notice = gr.HTML('', visible=False)
         with gr.Row():
             with gr.Column(scale=1):
                 novita_key = gr.Textbox(value="", label="Novita.AI API KEY (store in broweser)", placeholder="novita.ai api key", type="password")
@@ -191,7 +193,8 @@ def create_ui():
                     _hide_lora_training_response = gr.JSON(visible=False)
                     style_height = gr.Slider(minimum=1, maximum=1024, step=1, label="Style Height", value=512)
                     style_width = gr.Slider(minimum=1, maximum=1024, step=1, label="Style Width", value=512)
-                    style_method = gr.Radio(choices=["txt2img", "controlnet-depth", "controlnet-pose", "controlnet-canny", "controlnet-lineart", "controlnet-scribble", "controlnet-tile"], label="Style Method")
+                    style_method = gr.Radio(choices=["txt2img", "controlnet-depth", "controlnet-pose", "controlnet-canny",
+                                            "controlnet-lineart", "controlnet-scribble", "controlnet-tile"], label="Style Method")
 
                     style_advanced = gr.Checkbox(label="Advanced")
                     with gr.Column(scale=1, visible=False) as style_advanced_tab:
@@ -587,22 +590,26 @@ def create_ui():
 
         def onload(novita_key):
             if novita_key is None or novita_key == "":
-                return novita_key, gr.update(choices=[], value=None), gr.update(value=None), f"$ UNKNOWN"
+                return novita_key, gr.update(choices=[], value=None), gr.update(value=None), f"$ UNKNOWN", gr.update(visible=False)
             try:
                 user_info_json = get_noviata_client(novita_key).user_info()
                 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")]
             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=[], value=None), gr.update(value=None), f"$ UNKNOWN", gr.update(visible=False)
             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}"
+            free_trial = user_info_json.free_trial.get('training', 0)
+            trial_html = f'''<h2 style="color: red"> 🌟 Free trial quota: {free_trial} </h2>'''
+
+            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}", gr.update(value=trial_html, visible=free_trial > 0)
 
-        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]; }")
+        novita_key.change(onload, inputs=novita_key, outputs=[novita_key, style_lora, _hide_lora_training_response,
+                          user_balance, free_trial_notice], _js="(v)=>{ setStorage('novita_key',v); return [v]; }")
 
         demo.load(
             inputs=[novita_key],
-            outputs=[novita_key, style_lora, _hide_lora_training_response, user_balance],
+            outputs=[novita_key, style_lora, _hide_lora_training_response, user_balance, free_trial_notice],
             fn=onload,
             _js=get_local_storage,
         )

requirements.txt

@@ -1,2 +1,2 @@
-novita_client==0.4.5
+novita_client==0.4.6
 gradio==3.50.2