Update app.py

app.py

@@ -14,8 +14,6 @@ def remove_space(example):
     return {'Corpus-Delicti': example['Corpus-Delicti'].replace(" ", "").split('一、')[1]} 
 
 def download_file(content, filename):
-#    print(filename)
-#    print(content)
     with open(filename, "w", encoding="utf-8") as f:
         f.write(content)
 
@@ -39,8 +37,8 @@ def random_next(num_examples=5):
     court_name = random_selected['Court'][0]
     case_no = random_selected['CaseNo'][0]
     crime_descrip = random_selected['Corpus-Delicti'][0]
-    title = court_name + "_" + case_no
-    data_tuple = (court_name, case_no, crime_descrip, title)
+    filename = court_name + "_" + case_no + '.txt'
+    data_tuple = (court_name, case_no, crime_descrip, filename)
     return data_tuple
 
 def gen_template(crime_descrip, element, tag):
@@ -75,12 +73,13 @@ use_auth_token = os.environ['HUB_TOKEN'] # 下載判決書資料集所需要的
 login(token = os.environ['HUB_TOKEN'])
 dataset = load_dataset("jslin09/Fraud_Case_Verdicts", use_auth_token=use_auth_token, revision="main")
 dataset = dataset.map(remove_space)
-#random_selected = random_elements(dataset["train"])
+
+# 隨機選取案件
 random_selected = random_next()
 court_name = random_selected[0]
 case_no = random_selected[1]
 crime_descrip = random_selected[2]
-title = random_selected[3]
+filename = random_selected[3]
 
 with gr.Blocks() as demo:
     gr.Markdown(
@@ -89,22 +88,17 @@ with gr.Blocks() as demo:
     """)
     with gr.Row():
         with gr.Column(): # 犯罪事實段
-#            court_name = random_selected[0]
-#            case_no = random_selected[1]
-#            crime_descrip = random_selected[2]
             with gr.Row(): # 抬頭段
                 courtName = gr.Label(label='法院名稱', value=court_name, visible=False)
                 caseNo = gr.Label(label='案號', value=case_no, visible=False)
-                title = gr.components.Textbox(label='案號',value=title)
+                filename = gr.components.Textbox(label='案號',value=filename, show_copy_button=True)
             prompt = gr.components.Textbox(lines=5, label='犯罪事實',value=crime_descrip)
             with gr.Row():
                 with gr.Column():
                     btn = gr.Button("隨機選擇")
-#            gr.Examples(examples, inputs=[prompt])
         with gr.Column():
             with gr.Row():
                 element = gr.components.Textbox(lines=2, label="構成要件要素")
-#                tag = gr.components.Textbox(label="標籤")
             tag = gr.Dropdown(
                               choices = [("被告(犯罪主體)","<LEO_SOC>"), ("主觀犯意", "<LEO_SLE>"), ("不法行為","<LEO_ACT>"), ("因果關係","<LEO_CAU>"),
                                          ("被害人/告訴人","<LEO_VIC>"), ("危害結果","<LEO_ROH>"), ("未遂","<LEO_ATP>"), ("既遂","<LEO_ACC>"),
@@ -115,7 +109,7 @@ with gr.Blocks() as demo:
                     btn2 = gr.Button("產生標註語料內容")
             result = gr.components.Textbox(lines=5, label="語料內容", show_copy_button=True)
 #            btn3 = gr.Button("下載")
-        btn.click(random_next, inputs=[], outputs=[courtName, caseNo, prompt, title])
+        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=[])