Update app.py

app.py

@@ -68,6 +68,23 @@ def gen_template(crime_descrip, element, tag):
     template = blurb + '\n' + descript + '\n' + element + '\n' + tag + '\n' + end
     return template
 
+js_download = 'function downloadFile(result, filename) {
+  //藉型別陣列建構的 blob 來建立 URL
+  let fileName = filename;
+  const data = result;
+  let blob = new Blob([data], {
+    type: "application/octet-stream",
+  });
+  var href = URL.createObjectURL(blob);
+  // 從 Blob 取出資料
+  var link = document.createElement("a");
+  document.body.appendChild(link);
+  link.href = href;
+  link.download = fileName;
+  link.click();
+}
+'
+
 # 下載判決書資料集
 use_auth_token = os.environ['HUB_TOKEN'] # 下載判決書資料集所需要的 token。
 login(token = os.environ['HUB_TOKEN'])
@@ -108,10 +125,10 @@ with gr.Blocks() as demo:
                 with gr.Column():
                     btn2 = gr.Button("產生標註語料內容")
             result = gr.components.Textbox(lines=5, label="語料內容", show_copy_button=True)
-#            btn3 = gr.Button("下載")
+            btn3 = gr.Button("下載")
         btn.click(random_next, inputs=[], outputs=[courtName, caseNo, prompt, filename])
         btn2.click(gen_template, inputs=[prompt, element, tag], outputs=[result])
-#        btn3.click(download_file, inputs=[result, title], outputs=[])
+        btn3.click(None, inputs=[result, filename], js_=js_download)
 
 if __name__ == "__main__":
     demo.launch() # 在遠端啟動時，需要 share=True 。