download button

app.py

@@ -36,7 +36,7 @@ def process(input_bib, shorten, remove_keys, deduplicate, sort):
         output_bib = f.read().replace("\n ", "\n    ")
     # delete both files
     # print(output_bib)
-    return output_bib
+    return output_bib, random_id
 
 
 example_input = """
@@ -67,7 +67,6 @@ with gr.Blocks() as demo:
     
     gr.Markdown(
             '''# Rebiber: A tool for normalizing bibtex with official info.
-
                 <table> 
                 <tr>
                 <td>
@@ -87,11 +86,10 @@ with gr.Blocks() as demo:
                 </td>
                 </tr>
                 </table>
-
+                
                 We often cite papers using their arXiv versions without noting that they are already __PUBLISHED__ in some conferences. These unofficial bib entries might violate rules about submissions or camera-ready versions for some conferences. 
                 We introduce __Rebiber__, a simple tool in Python to fix them automatically. It is based on the official conference information from the [DBLP](https://dblp.org/) or [the ACL anthology](https://www.aclweb.org/anthology/) (for NLP conferences)!
-                Apart from handling outdated arXiv citations, __Rebiber__ also normalizes citations in a unified way (DBLP-style), supporting abbreviation and value selection.
-                Note that you can copy the output bib file by clicking the top-right button.
+                Apart from handling outdated arXiv citations, __Rebiber__ also normalizes citations in a unified way (DBLP-style), supporting abbreviation and value selection.  
             '''
     )
     
@@ -101,14 +99,30 @@ with gr.Blocks() as demo:
             removekeys = gr.CheckboxGroup(["url", "biburl", "address", "publisher", "pages", "doi", "volume", "bibsource"], 
                                 value=[False, False, False, False, False, False, False, False],
                                 label="Remove Keys", info="Which keys to remove?")
-            shorten = gr.Checkbox(label="Abbreviation", info="Shorten the conference/journal names (e.g., `Proceedings of the 2020 International Conference of ...` --> `Proc. of ICML')", value=False)
+            shorten = gr.Checkbox(label="Abbreviation", info="Shorten the conference/journal names (e.g., `Proceedings of the 2020 International Conference of ...` --> `Proc. of ICML')", value=False)            
             dedup = gr.Checkbox(label="Deduplicate entries.", value=False)
             sort = gr.Checkbox(label="Sort alphabetically by ID.", value=False)
-            button = gr.Button("Submit")
+            with gr.Row():
+                clr_button = gr.Button("Clear")
+                button = gr.Button("Submit")
+            ex_uuid = gr.Text(label="UUID")
+            ex_uuid.visible = False
         with gr.Column(scale=3):
-            output=gr.outputs.Textbox(label="Output BIB").style(show_copy_button=True, interactive=True)        
-    button.click(process, inputs=[input_bib, shorten, removekeys, dedup, sort], outputs=output, api_name = "process")
-    
+            output=gr.Textbox(label="Output BIB (Note that you can copy the output bib file by clicking the top-right button.)").style(show_copy_button=True, interactive=False)        
+            download_btn = gr.Button("Download")
+            download_content = gr.outputs.File()
+            download_content.visible = False 
+    def download_file(ex_uuid):
+        global download_content
+        # Replace this with your code to generate/download the file
+        file_path = f"output_{ex_uuid}.bib"
+        download_content.update(visible=False)
+        return file_path, gr.update(visible=True)
+    download_btn.click(download_file, inputs=ex_uuid, outputs=[download_content,download_content])  
+    button.click(process, inputs=[input_bib, shorten, removekeys, dedup, sort], outputs=[output, ex_uuid], api_name = "process")
+    def clean(text):
+        return ""
+    clr_button.click(clean, input_bib, input_bib)
     # gr.Interface(fn=process,
     # outputs=gr.outputs.Textbox(label="Output BIB").style(show_copy_button=True),
     # examples=examples,