Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import json
+import os
 
 # Lista para armazenar os dados
 data = []
@@ -24,13 +25,14 @@ def add_entry(context, question, answer, answer_start):
     data.append(entry)
     return "Entry added!"
 
-# Função para salvar os dados em um arquivo JSON
+# Função para salvar os dados em um arquivo JSON e retornar o caminho do arquivo
 def save_to_json():
     global data
     squad_format = {"data": data}
-    with open("squad_dataset.json", "w", encoding='utf-8') as f:
+    file_path = "squad_dataset.json"
+    with open(file_path, "w", encoding='utf-8') as f:
         json.dump(squad_format, f, ensure_ascii=False, indent=4)
-    return "Data saved to squad_dataset.json"
+    return file_path
 
 # Interface Gradio
 with gr.Blocks() as demo:
@@ -42,8 +44,9 @@ with gr.Blocks() as demo:
     add_button = gr.Button("Add Entry")
     save_button = gr.Button("Save to JSON")
     output = gr.Textbox(label="Output")
+    download_link = gr.File(label="Download JSON", interactive=False)
 
     add_button.click(fn=add_entry, inputs=[context, question, answer, answer_start], outputs=output)
-    save_button.click(fn=save_to_json, outputs=output)
+    save_button.click(fn=save_to_json, outputs=download_link)
 
 demo.launch()