Upload app.py

app.py

@@ -2,44 +2,46 @@ import gradio as gr
 import json
 import os
 
-# JSON file to store data
+# Define JSON storage file
 DATA_FILE = "queries.json"
 
-# Load existing data if available
-if os.path.exists(DATA_FILE):
-    with open(DATA_FILE, "r") as file:
-        stored_data = json.load(file)
-else:
-    stored_data = []
+# Ensure the file exists
+if not os.path.exists(DATA_FILE):
+    with open(DATA_FILE, "w") as file:
+        json.dump([], file)  # Initialize empty list
+
+# Load existing data
+with open(DATA_FILE, "r") as file:
+    stored_data = json.load(file)
+
+# Function to add bulk queries
+def add_bulk_queries(category, queries):
+    global stored_data
+    queries_list = queries.split("\n")  # Split input by new lines
 
-# Function to add a new query
-def add_query(category, query):
-    new_entry = {"Category": category, "Query": query}
-    stored_data.append(new_entry)
+    for query in queries_list:
+        if query.strip():  # Avoid empty lines
+            stored_data.append({"Category": category, "Query": query.strip()})
 
     # Save to JSON file
     with open(DATA_FILE, "w") as file:
         json.dump(stored_data, file, indent=4)
 
-    return f"Added: {category} - {query}"
+    return f"Added {len(queries_list)} queries under category '{category}'"
 
-# Function to download the stored dataset
-def download_json():
-    return DATA_FILE
-
-# Gradio Interface
+# Gradio UI
 with gr.Blocks() as app:
-    gr.Markdown("# 📝 Query Collection App")
-    gr.Markdown("### Manually enter queries to store in a JSON file:")
+    gr.Markdown("# 📝 Bulk Query Input & Storage")
+    gr.Markdown("### Paste multiple queries and store them in a JSON file.")
 
-    category_input = gr.Textbox(label="Category")
-    query_input = gr.Textbox(label="Query")
-    submit_button = gr.Button("Add Query")
-    status_output = gr.Textbox(label="Status", interactive=False)
+    category_input = gr.Textbox(label="Category", placeholder="Enter category (e.g., Rhyme, Aviation and Math)")
+    query_input = gr.Textbox(label="Queries (one per line)", lines=10, placeholder="Paste queries here...")
+    submit_button = gr.Button("Add Queries")
 
-    submit_button.click(add_query, inputs=[category_input, query_input], outputs=status_output)
+    status_output = gr.Textbox(label="Status", interactive=False)
+    submit_button.click(add_bulk_queries, inputs=[category_input, query_input], outputs=status_output)
 
-    gr.Markdown("### Download the stored dataset:")
+    gr.Markdown("### Download Stored Queries:")
     download_button = gr.File(DATA_FILE, label="Download JSON")
 
 app.launch()