Update app.py

app.py

@@ -1,26 +1,21 @@
 import gradio as gr
 import json
-import os
 
 # Lista para armazenar os dados
 data = []
 
 # Função para adicionar contextos, perguntas e respostas
-def add_entry(context, question, answer, answer_start):
+def add_entry(title, context, question, answer, answer_start, entry_id):
     global data
     entry = {
+        "title": title,
         "context": context,
-        "qas": [
-            {
-                "question": question,
-                "answers": [
-                    {
-                        "text": answer,
-                        "answer_start": answer_start
-                    }
-                ]
-            }
-        ]
+        "question": question,
+        "id": entry_id,
+        "answers": {
+            "answer_start": [answer_start],
+            "text": [answer]
+        }
     }
     data.append(entry)
     return "Entry added!"
@@ -28,25 +23,26 @@ def add_entry(context, question, answer, answer_start):
 # 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}
     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)
+        json.dump(data, f, ensure_ascii=False, indent=4)
     return file_path
 
 # 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")
     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)
+    add_button.click(fn=add_entry, inputs=[title, context, question, answer, answer_start, entry_id], outputs=output)
     save_button.click(fn=save_to_json, outputs=download_link)
 
 demo.launch()