Upload app.py

app.py

@@ -14,32 +14,50 @@ if not os.path.exists(DATA_FILE):
 with open(DATA_FILE, "r") as file:
     stored_data = json.load(file)
 
-# Function to add bulk queries
-def add_bulk_queries(category, queries):
+# Function to add a new query with positive and negative responses
+def add_query(category, query, positive1, positive2, negative1, negative2):
     global stored_data
-    queries_list = queries.split("\n")  # Split input by new lines
 
-    for query in queries_list:
-        if query.strip():  # Avoid empty lines
-            stored_data.append({"Category": category, "Query": query.strip()})
+    # Create structured entry
+    new_entry = {
+        "Category": category,
+        "Query": query,
+        "Positive_1": positive1,
+        "Positive_2": positive2,
+        "Negative_1": negative1,
+        "Negative_2": negative2
+    }
+
+    stored_data.append(new_entry)
 
     # Save to JSON file
     with open(DATA_FILE, "w") as file:
         json.dump(stored_data, file, indent=4)
 
-    return f"Added {len(queries_list)} queries under category '{category}'"
+    return f"Added query: {query} under category '{category}' with responses."
 
 # Gradio UI
 with gr.Blocks() as app:
-    gr.Markdown("# 📝 Bulk Query Input & Storage")
-    gr.Markdown("### Paste multiple queries and store them in a JSON file.")
+    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="Queries (one per line)", lines=10, placeholder="Paste queries here...")
-    submit_button = gr.Button("Add Queries")
+    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)
 
+    submit_button = gr.Button("Add Query")
     status_output = gr.Textbox(label="Status", interactive=False)
-    submit_button.click(add_bulk_queries, inputs=[category_input, query_input], outputs=status_output)
+
+    submit_button.click(
+        add_query, 
+        inputs=[category_input, query_input, positive1_input, positive2_input, negative1_input, negative2_input], 
+        outputs=status_output
+    )
 
     gr.Markdown("### Download Stored Queries:")
     download_button = gr.File(DATA_FILE, label="Download JSON")