Update app.py

app.py

@@ -18,13 +18,13 @@ def get_count():
     return "Total number of today's image generation: " + str(today_count)
 
 def control_ui():
-    prompt = gr.TextArea(label="Prompt", lines=3, value="{breast expansion},{gigantic breasts},[artist:ningen_mame],artist:ciloranko,[artist:sho_(sho_lwlw)],[artist:foifoi (marfoyfoyfoy)],1girl,skinny,narrow waist,")
+    prompt = gr.TextArea(elem_id='txt2img_prompt', label="Prompt", lines=3, value="{breast expansion},{gigantic breasts},[artist:ningen_mame],artist:ciloranko,[artist:sho_(sho_lwlw)],[artist:foifoi (marfoyfoyfoy)],1girl,skinny,narrow waist,")
     quality_tags = gr.TextArea(
-        label="Quality Tags", lines=1,
+        elem_id='txt2img_qua_prompt', label="Quality Tags", lines=1,
         value="best quality, amazing quality, very aesthetic, absurdres",
     )
     neg_prompt = gr.TextArea(
-        label="Negative Prompt", lines=1,
+        elem_id='txt2img_neg_prompt', label="Negative Prompt", lines=1,
         value="wide hips,thick thighs,plump,lowres, {bad}, error, fewer, extra, missing, worst quality, jpeg artifacts, bad quality, watermark, unfinished, displeasing, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract],pubic hair",
     )
     with gr.Row():
@@ -144,7 +144,10 @@ def util_ui():
     return page, png2main, items, info, image
 
 def ui():
-    with gr.Blocks(title="NAI Client") as website:
+    head = ''
+    for f in sorted(os.listdir('./tagcomplete/javascript')):
+        head += f'<script type="text/javascript" src="file=tagcomplete/javascript/{f}"></script>\n'
+    with gr.Blocks(title="NAI Client", head=head) as website:
         with gr.Tabs():
             with gr.TabItem("Main", elem_id="client_ui_main"):
                 _, paras = main_ui()
@@ -163,4 +166,4 @@ def ui():
 if __name__ == '__main__':
     website = ui()
     website.queue()
-    website.launch(auth=(os.environ.get('account'), os.environ.get('password')))
+    website.launch(auth=(os.environ.get('account'), os.environ.get('password')), allowed_paths=['tagcomplete'])