Update app/webui/app.py

app/webui/app.py

@@ -114,6 +114,13 @@ def switch(source_lang,source_text,target_lang,output_final):
     else:
         return gr.update(value=target_lang), gr.update(value=source_text), gr.update(value=source_lang), gr.update(value="")
 
+def closeBtnShow():
+    return gr.update(visible=False), gr.update(visible=True)
+    
+def closeBtnHide(output_final):
+    if output_final:
+        return gr.update(visible=True), gr.update(visible=False)
+
 TITLE = """
     <div style="display: inline-flex;">
         <div style="margin-left: 6px; font-size:32px; color: #6366f1"><b>Translation Agent</b> WebUI</div>
@@ -265,14 +272,24 @@ with gr.Blocks(theme="soft", css=CSS, fill_height=True) as demo:
         upload = gr.UploadButton(label="Upload", file_types=["text"])
         export = gr.DownloadButton(visible=False)
         clear = gr.ClearButton([source_text, output_init, output_reflect, output_final])
+        close = gr.Button(value="Stop", visible=False)
 
     switchBtn.click(fn=switch, inputs=[source_lang,source_text,target_lang,output_final], outputs=[source_lang,source_text,target_lang,output_final])
+    
     menuBtn.click(fn=update_menu, inputs=visible, outputs=[visible, menubar], js=JS)
     endpoint.change(fn=update_model, inputs=[endpoint], outputs=[model])
+    
     choice.select(fn=enable_sec, inputs=[choice], outputs=[endpoint2, model2, api_key2])
     endpoint2.change(fn=update_model, inputs=[endpoint2], outputs=[model2])
+    
     submit.click(fn=huanik, inputs=[endpoint, model, api_key, choice, endpoint2, model2, api_key2, source_lang, target_lang, source_text, country, max_tokens, context_window, num_output, rpm], outputs=[output_init, output_reflect, output_final, output_diff])
+    
     upload.upload(fn=read_doc, inputs = upload, outputs = source_text)
     output_final.change(fn=export_txt, inputs=output_final, outputs=[export])
+    
+    submit.click(fn=closeBtnShow, outputs=[clear, close])
+    output_final.change(fn=closeBtnHide, inputs=output_final, outputs=[clear, close])
+    close.click(fn=None, cancels=start_ta)
+    
 if __name__ == "__main__":
     demo.queue(api_open=False).launch(show_api=False, share=False)