Add thumbs up (#22)

* add like button; log response to mongodb

* remove DEV from space title

README.md

@@ -1,5 +1,5 @@
 ---
-title: TowardsAI 🤝 Buster - Dev
+title: TowardsAI 🤝 Buster
 emoji: 🤖
 colorFrom: blue
 colorTo: blue

app.py

@@ -35,6 +35,24 @@ AVAILABLE_SOURCES = [
 ]
 
 
+def log_likes(completion: Completion, like_data: gr.LikeData):
+    # make it a str so json-parsable
+
+    collection = "liked_data-test"
+
+    completion_json = completion.to_json(
+        columns_to_ignore=["embedding", "similarity", "similarity_to_answer"]
+    )
+    completion_json["liked"] = like_data.liked
+    logger.info(f"User reported {like_data.liked=}")
+
+    try:
+        cfg.mongo_db[collection].insert_one(completion_json)
+        logger.info("")
+    except:
+        logger.info("Something went wrong logging")
+
+
 def format_sources(matched_documents: pd.DataFrame) -> str:
     if len(matched_documents) == 0:
         return ""
@@ -132,6 +150,9 @@ with demo:
         gr.Markdown(
             "<h3><center>Toward's AI x Buster 🤖: A Question-Answering Bot for anything AI-related</center></h3>"
         )
+
+    latest_completion = gr.State()
+
     source_selection = gr.Dropdown(
         choices=AVAILABLE_SOURCES_UI,
         label="Select Sources",
@@ -159,15 +180,16 @@ with demo:
         "\n\n### Powered by [Buster 🤖](www.github.com/jerpint/buster)"
     )
 
-    response = gr.State()
+    completion = gr.State()
 
     submit.click(user, [question, chatbot], [question, chatbot], queue=False).then(
-        get_answer, inputs=[chatbot, source_selection], outputs=[chatbot, response]
-    ).then(add_sources, inputs=[chatbot, response], outputs=[chatbot])
+        get_answer, inputs=[chatbot, source_selection], outputs=[chatbot, completion]
+    ).then(add_sources, inputs=[chatbot, completion], outputs=[chatbot])
     question.submit(user, [question, chatbot], [question, chatbot], queue=False).then(
-        get_answer, inputs=[chatbot, source_selection], outputs=[chatbot, response]
-    ).then(add_sources, inputs=[chatbot, response], outputs=[chatbot])
+        get_answer, inputs=[chatbot, source_selection], outputs=[chatbot, completion]
+    ).then(add_sources, inputs=[chatbot, completion], outputs=[chatbot])
 
+    chatbot.like(log_likes, completion)
 
 demo.queue(concurrency_count=16)
 demo.launch(debug=True, share=False)

cfg.py

@@ -9,6 +9,12 @@ from buster.retriever import DeepLakeRetriever, Retriever
 from buster.tokenizers import GPTTokenizer
 from buster.validators import QuestionAnswerValidator, Validator
 
+from utils import init_mongo_db
+
+MONGODB_URI = os.getenv("MONGODB_URI")
+mongo_db = init_mongo_db(uri=MONGODB_URI, db_name="towardsai-buster")
+
+
 logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.INFO)
 

utils.py

@@ -0,0 +1,16 @@
+from pymongo.mongo_client import MongoClient
+from pymongo.server_api import ServerApi
+
+
+def init_mongo_db(uri: str, db_name: str):
+    """Initialize the mongodb database."""
+
+    try:
+        assert uri is not None, "No URI passed"
+        client = MongoClient(uri, server_api=ServerApi("1"))
+        database = client[db_name]
+        print("Connected to MongoDB")
+        return database
+    except Exception as e:
+        print("Something went wrong connecting to mongodb")
+        return