Update app.py

app.py

@@ -269,10 +269,8 @@ def main(input_article, input_trigger):
     # return updated_article, modified, merged_par_pred_df
     modified_in_all = str(len(paragraphs_needed)) + " / " + str(len(data_test))
     
-    try:
-        os.remove("./util/experiments/classification.csv")
-        os.remove("./util/experiments/paragraphs_needed.csv")
-    except: pass
+    os.remove("./util/experiments/classification.csv")
+    os.remove("./util/experiments/paragraphs_needed.csv")
     return updated_article, modified_in_all
 
 def copy_to_clipboard(t):
@@ -284,37 +282,10 @@ def compare_versions():
     old, new = [], []
     with open("./util/experiments/formatted_input.txt", "r") as f:
         old = f.read()
+        old = old.replace("[ADD]", "")
     with open("./util/experiments/updated_article.txt", "r") as f:
         new = f.read()
     return old, new
-"""
-demo = gr.Interface(
-	main,
-	[
-    	gr.Textbox(
-            lines=2, label="Non-updated Article", placeholder="Input the article..."
-        ),
-        gr.Textbox(
-            lines=2, label="Triggered News Event", placeholder="Input the triggered news event..."
-        )
-    ],
-	[
-        gr.Textbox(
-            lines=25,
-            label="Output",
-        ),
-        gr.Textbox(
-            lines=1,
-            label="#MODIFIED/ALL"
-        ),
-        # btn = gr.Button(value="Copy Updated Article to Clipboard")
-        # btn.click(copy_to_clipboard)
-        # gr.components.Button(value="Copy Updated Article to Clipboard", fn=copy_to_clipboard),
-    ],
-    title="Event Triggered Article Updating System",
-    description="Powered by YTLee",
-)
-"""
 
 with open("./examples/non_update.txt", "r") as f:
     exin_1 = f.read()
@@ -342,7 +313,7 @@ with gr.Blocks() as demo:
         )
     with gr.Tab("Article Updating"):
         input_1 = gr.Textbox(label="Non-updated Article", lines=2, placeholder="Input the contexts...")
-        input_2 = gr.Textbox(label="Triggered News Event", lines=2, placeholder="Input the triggered news event...") 
+        input_2 = gr.Textbox(label="Triggered News Event", lines=1, placeholder="Input the triggered news event...") 
         btn = gr.Button(value="Submit")
         with gr.Row():
             output_1 = gr.Textbox(label="Updated Article", lines=5)
@@ -367,12 +338,10 @@ with gr.Blocks() as demo:
             com_1 = gr.Textbox(label="Non-update Article", value=com_1_value, lines=15)
             com_2 = gr.Textbox(label="Updated Article", value=com_2_value, lines=15)
         btn_com.click(fn=compare_versions, inputs=[], outputs=[com_1, com_2])
-        path_formatted_input = "./util/experiments/formatted_input.txt"
-        path_updated_article = "./util/experiments/updated_article.txt" 
-        if os.path.isfile(path_formatted_input):
-            os.remove(path_formatted_input)
-        if os.path.isfile(path_updated_article):
-            os.remove(path_updated_article)
+        formatted_input_path = "./util/experiments/formatted_input.txt"
+        updated_article_path = "./util/experiments/updated_article.txt"
+        if os.path.isfile(formatted_input_path): os.remove(formatted_input_path)
+        if os.path.isfile(updated_article_path): os.remove(updated_article_path)
     gr.HTML("""
             <div align="center">
                 <p>