update

app.py

@@ -1,13 +1,10 @@
 import gradio as gr
 
-def origin_preditcion(title, context):
-    return f"Title:{title}\nContext:{context}\n..."
-
 def edit_process(title, context):
     return f"Title:{title}\nContext:{context}\n...", f"Title:{title}\nContext:{context}\n..."
 
 def add_process(title, context, img):
-    return f"Title:{title}\nContext:{context}\n...{img}"
+    return f"Title:{title}\nContext:{context}\n...{img}", f"Title:{title}\nContext:{context}\n...{img}"
 
 with gr.Blocks() as demo:
     gr.Markdown("# KGE Editing")
@@ -19,7 +16,7 @@ with gr.Blocks() as demo:
             with gr.Row():
                 with gr.Column():
                     input = gr.Textbox(label="Input", lines=1, placeholder="Mask triple input")
-                    origin_button = gr.Button("Origin")
+                    # origin_button = gr.Button("Origin")
 
                 with gr.Column():
                     origin_output = gr.Textbox(label="Before Edit", lines=3, placeholder="")
@@ -48,18 +45,19 @@ with gr.Blocks() as demo:
                     add_button = gr.Button("Add")
 
                 with gr.Column():
-                    add_output = gr.Textbox(label="Add Results", lines=5, placeholder="")
+                    add_origin_output = gr.Textbox(label="Origin Results", lines=3, placeholder="")
+                    add_output = gr.Textbox(label="Add Results", lines=3, placeholder="")
 
             gr.Examples(
                 examples=[["h1 r1 t1", "1"], ["h2 r2 t2", "1"]],
                 inputs=[input, alter_label],
-                outputs=add_output,
+                outputs=[add_origin_output, add_output],
                 fn=add_process,
                 cache_examples=True,
             )
 
-    origin_button.click(fn=origin_preditcion, inputs=[input, alter_label], outputs=origin_output)
+    # origin_button.click(fn=origin_preditcion, inputs=[input, alter_label], outputs=origin_output)
     edit_button.click(fn=edit_process, inputs=[input, alter_label], outputs=[origin_output, edit_output])
-    add_button.click(fn=add_process, inputs=[input, alter_label], outputs=add_output)
+    add_button.click(fn=add_process, inputs=[input, alter_label], outputs=[add_origin_output, add_output])
 
 demo.launch()