Update heights

app.py

@@ -113,16 +113,18 @@ def decode(paragraphs_needed):
 def split_article(article, trigger):
     if article.split("\n"): article = article.replace("\n", "\\\\c\\\\c")
     paragraphs = article.replace("\\c\\c", "\c\c").split("\\\\c\\\\c")
+    format_pars = [par for par in paragraphs]
     pars = [str(par) + " -- " + str(trigger) for par in paragraphs]
     # pd.DataFrame({"paragraph": pars}).to_csv("./util/experiments/input_paragraphs.csv")
-    return pars 
+    formatted_input = "\n".join(format_pars)
+    return pars, formatted_input 
 
 def config():
     load_dotenv()
 
 def call_gpt(paragraph, trigger):
     # openai.api_key = os.environ.get("GPT_API")
-    openai.api_key = "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"
+    openai.api_key = "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"
     tokenizer = BartTokenizer.from_pretrained("theQuert/NetKUp-tokenzier")
     inputs_for_gpt = f"""
 As an article writer, your task is to provide an updated paragraph in the length same as non-updated paragraph based on the given non-updated paragraph and a triggered news.
@@ -163,9 +165,9 @@ As an article writer, your task is to provide an updated paragraph in the length
 
     
 def main(input_article, input_trigger):
-    # csv_path = "./util/experiments/input_paragraphs.csv"
-    # if os.path.isfile(csv_path):
-    #     os.remove(csv_path)
+    csv_path = "./util/experiments/input_paragraphs.csv"
+    if os.path.isfile(csv_path):
+        os.remove(csv_path)
     modified = "TRUE"
     # device = "cuda" if torch.cuda.is_available() else "cpu"
     device="cpu"
@@ -179,7 +181,9 @@ def main(input_article, input_trigger):
                     )
 
     # split the input article to paragraphs in tmp csv format
-    data_test = split_article(input_article, input_trigger)
+    data_test, formatted_input = split_article(input_article, input_trigger)
+    with open("./util/experiments/formatted_input.txt", "w") as f:
+        f.write(formatted_input)
 
     seed_val = 42
     random.seed(seed_val)
@@ -264,6 +268,11 @@ def main(input_article, input_trigger):
     merged_par_pred_df = pd.DataFrame({"paragraphs": data_test, "predictions": predictions}).to_csv("./util/experiments/par_with_class.csv")
     # 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
     return updated_article, modified_in_all
 
 def copy_to_clipboard(t):
@@ -271,6 +280,13 @@ def copy_to_clipboard(t):
         t = f.read()
         pyperclip.copy(t)
 
+def compare_versions():
+    old, new = [], []
+    with open("./util/experiments/formatted_input.txt", "r") as f:
+        old = f.read()
+    with open("./util/experiments/updated_article.txt", "r") as f:
+        new = f.read()
+    return old, new
 """
 demo = gr.Interface(
 	main,
@@ -329,25 +345,34 @@ with gr.Blocks() as demo:
         input_2 = gr.Textbox(label="Triggered News Event", lines=2, placeholder="Input the triggered news event...") 
         btn = gr.Button(value="Submit")
         with gr.Row():
-            output_1 = gr.Textbox(label="Updated Article", lines=10)
+            output_1 = gr.Textbox(label="Updated Article", lines=5)
             output_2 = gr.Textbox(label="#MODIFIED / #ALL")
         btn.click(fn=main, inputs=[input_1, input_2], outputs=[output_1, output_2])
         btn_copy = gr.Button(value="Copy Updated Article to Clipboard")
         btn_copy.click(fn=copy_to_clipboard, inputs=[output_1], outputs=[])
         gr.Markdown("## Input Examples")
+        gr.Markdown("### There are 2 examples below, click them to test inputs automatically!")
         gr.Examples(
             examples=[[exin_1, trigger_1], [exin_2, trigger_2]],
             fn=main,
             inputs=[input_1, input_2],
             outputs=[output_1, output_2],
-            # cache_examples=True
+            # cache_examples=True,
+            # run_on_click=True,
                 ),
+        com_1_value, com_2_value = "Pls finish article updating, then click the button above", "Pls finish article updating, then click the button above."
     with gr.Tab("Compare between versions"):
         btn_com = gr.Button(value="Differences Highlighting")
-        btn_com.click(fn=copy_to_clipboard, inputs=[output_1], outputs=[])
         with gr.Row():
-            com_1 = gr.Textbox(label="Non-update Article", lines=25)
-            com_2 = gr.Textbox(label="Updated Article", lines=25)
+            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)
     gr.HTML("""
             <div align="center">
                 <p>