Upload app.py

app.py

@@ -34,7 +34,12 @@ def add_query(category, query, positive1, positive2, negative1, negative2):
     with open(DATA_FILE, "w") as file:
         json.dump(stored_data, file, indent=4)
 
-    return f"Added query: {query} under category '{category}' with responses."
+    return f"✅ Added query: {query} under category '{category}' with responses."
+
+# Function to serve the updated JSON file for download
+def download_json():
+    with open(DATA_FILE, "r") as file:
+        return file.read()
 
 # Gradio UI
 with gr.Blocks() as app:
@@ -59,7 +64,10 @@ with gr.Blocks() as app:
         outputs=status_output
     )
 
-    gr.Markdown("### Download Stored Queries:")
-    download_button = gr.File(DATA_FILE, label="Download JSON")
+    gr.Markdown("### Save & Download Stored Queries:")
+    download_button = gr.Button("Download JSON")
+    download_output = gr.File(label="Download File")
+
+    download_button.click(download_json, outputs=download_output)
 
 app.launch()