Allow user to download history(accept/reject) #44

Author: pramitchoudhary

app.toml

@@ -4,7 +4,7 @@ title = "SQL-Sidekick"
 description = "QnA with tabular data using NLQ"
 LongDescription = "about.md"
 Tags = ["DATA_SCIENCE", "MACHINE_LEARNING", "NLP"]
-Version = "0.0.16"
+Version = "0.0.17"
 
 [Runtime]
 MemoryLimit = "64Gi"

ui/app.py

@@ -670,6 +670,31 @@ async def on_event(q: Q):
             _msg = "Sorry, try generating a conversation to save."
         q.page["chat_card"].data += [_msg, False]
         event_handled = True
+    elif q.args.download_accept:
+        result_path = f"{base_path}/var/lib/tmp/.cache/{q.user.table_name}/history.jsonl"
+        # Check if path exists
+        if Path(result_path).exists():
+            logging.info(f"Downloading accepted QnA history for table: {q.user.table_name}")
+            (server_path,) = await q.site.upload([result_path])
+            q.page["meta"].script = ui.inline_script(f'window.open("{server_path}", "_blank");')
+            os.remove(result_path)
+            _msg = "Download complete!"
+        else:
+            _msg = "No history found!"
+        q.page["chat_card"].data += [_msg, False]
+        event_handled = True
+    elif q.args.download_reject:
+        logging.info(f"Downloading rejected QnA history for table: {q.user.table_name}")
+        result_path = f"{base_path}/var/lib/tmp/.cache/{q.user.table_name}/invalid/history.jsonl"
+        if Path(result_path).exists():
+            (server_path,) = await q.site.upload([result_path])
+            q.page["meta"].script = ui.inline_script(f'window.open("{server_path}", "_blank");')
+            os.remove(result_path)
+            _msg = "Download complete!"
+        else:
+            _msg = "No history found!"
+        q.page["chat_card"].data += [_msg, False]
+        event_handled = True
     elif q.args.regenerate or q.args.regenerate_with_options:
         await chatbot(q)
         event_handled = True