Update app.py

app.py

@@ -26,7 +26,7 @@ def add_entry(context, question, answer, answer_start, entry_id):
     return "Entry added!"
 
 # Função para salvar os dados em um arquivo JSON e retornar o caminho do arquivo
-def save_to_json():
+def save_to_json(file_name):
     global data
     file_path = "squad_dataset.json"
     with open(file_path, "w", encoding='utf-8') as f:
@@ -36,17 +36,20 @@ def save_to_json():
 # Interface Gradio
 with gr.Blocks() as demo:
     with gr.Row():
+        title = gr.Textbox(label="Title")
         context = gr.Textbox(lines=5, label="Context")
         question = gr.Textbox(lines=2, label="Question")
         answer = gr.Textbox(lines=2, label="Answer")
         answer_start = gr.Number(label="Answer Start")
         entry_id = gr.Textbox(label="ID")
     add_button = gr.Button("Add Entry")
-    save_button = gr.Button("Save to JSON")
+    with gr.Row():
+        file_name = gr.Textbox(label="File Name", value="squad_dataset")
+        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, entry_id], outputs=output)
-    save_button.click(fn=save_to_json, outputs=download_link)
+    add_button.click(fn=add_entry, inputs=[title, context, question, answer, answer_start, entry_id], outputs=output)
+    save_button.click(fn=save_to_json, inputs=file_name, outputs=download_link)
 
 demo.launch()