add save_completion function (#46)

app.py

@@ -54,6 +54,19 @@ logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.INFO)
 
 
+def save_completion(completion: Completion, question: gr.Textbox):
+    collection = "completion_data-hf"
+
+    completion_json = completion.to_json(
+        columns_to_ignore=["embedding", "similarity", "similarity_to_answer"]
+    )
+    try:
+        cfg.mongo_db[collection].insert_one(completion_json)
+        logger.info("Completion saved to db")
+    except:
+        logger.info("Something went wrong logging completion to db")
+
+
 def log_likes(completion: Completion, like_data: gr.LikeData):
     collection = "liked_data-test"
 
@@ -217,10 +230,15 @@ with gr.Blocks(
 
     submit.click(user, [question, chatbot], [question, chatbot], queue=False).then(
         get_answer, inputs=[chatbot, source_selection], outputs=[chatbot, completion]
-    ).then(add_sources, inputs=[chatbot, completion], outputs=[chatbot])
+    ).then(add_sources, inputs=[chatbot, completion], outputs=[chatbot]).then(
+        save_completion, inputs=[completion]
+    )
+
     question.submit(user, [question, chatbot], [question, chatbot], queue=False).then(
         get_answer, inputs=[chatbot, source_selection], outputs=[chatbot, completion]
-    ).then(add_sources, inputs=[chatbot, completion], outputs=[chatbot])
+    ).then(add_sources, inputs=[chatbot, completion], outputs=[chatbot]).then(
+        save_completion, inputs=[completion]
+    )
 
     chatbot.like(log_likes, completion)