Upload app.py

app.py

@@ -11,14 +11,14 @@ if not os.path.exists(DATA_FILE):
         json.dump([], file)  # Initialize empty list
 
 # Load existing data
-with open(DATA_FILE, "r") as file:
-    stored_data = json.load(file)
+def load_data():
+    with open(DATA_FILE, "r") as file:
+        return json.load(file)
 
-# Function to add a new query with positive and negative responses
+# Function to add a new query
 def add_query(category, query, positive1, positive2, negative1, negative2):
-    global stored_data
+    data = load_data()  # Load latest data
 
-    # Create structured entry
     new_entry = {
         "Category": category,
         "Query": query,
@@ -28,32 +28,25 @@ def add_query(category, query, positive1, positive2, negative1, negative2):
         "Negative_2": negative2
     }
 
-    stored_data.append(new_entry)
+    data.append(new_entry)
 
-    # Save to JSON file
+    # Save updated data to JSON file
     with open(DATA_FILE, "w") as file:
-        json.dump(stored_data, file, indent=4)
-
-    return f"✅ Added query: {query} under category '{category}' with responses."
+        json.dump(data, file, indent=4)
 
-# Function to serve the updated JSON file for download
-def download_json():
-    with open(DATA_FILE, "r") as file:
-        return file.read()
+    return f"✅ Added query: {query} under category '{category}'"
 
 # Gradio UI
 with gr.Blocks() as app:
     gr.Markdown("# 📝 Query Storage App with Responses")
-    gr.Markdown("### Enter queries and their responses to store in a JSON file.")
-
-    category_input = gr.Textbox(label="Category", placeholder="Enter category (e.g., Rhyme, Aviation and Math)")
-    query_input = gr.Textbox(label="Query", placeholder="Enter query text", lines=2)
     
-    gr.Markdown("### Provide Responses")
-    positive1_input = gr.Textbox(label="Positive Response 1", placeholder="Enter first positive response", lines=3)
-    positive2_input = gr.Textbox(label="Positive Response 2", placeholder="Enter second positive response", lines=3)
-    negative1_input = gr.Textbox(label="Negative Response 1", placeholder="Enter first negative response", lines=3)
-    negative2_input = gr.Textbox(label="Negative Response 2", placeholder="Enter second negative response", lines=3)
+    category_input = gr.Textbox(label="Category", placeholder="Enter category")
+    query_input = gr.Textbox(label="Query", placeholder="Enter query text")
+    
+    positive1_input = gr.Textbox(label="Positive Response 1", placeholder="Enter first positive response")
+    positive2_input = gr.Textbox(label="Positive Response 2", placeholder="Enter second positive response")
+    negative1_input = gr.Textbox(label="Negative Response 1", placeholder="Enter first negative response")
+    negative2_input = gr.Textbox(label="Negative Response 2", placeholder="Enter second negative response")
 
     submit_button = gr.Button("Add Query")
     status_output = gr.Textbox(label="Status", interactive=False)
@@ -64,10 +57,7 @@ with gr.Blocks() as app:
         outputs=status_output
     )
 
-    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)
+    gr.Markdown("### Download Stored Queries:")
+    download_button = gr.File(DATA_FILE, label="Download JSON")
 
 app.launch()