change db to online db

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from response_db import StResponseDb
+from response_db import ResponseDb
 from create_cache import Game_Cache
 import numpy as np
 from PIL import Image
@@ -11,7 +11,7 @@ import uuid
 import nltk
 nltk.download('punkt')
 
-db = StResponseDb()
+db = ResponseDb()
 css = """
 .chatbot {display:flex;flex-direction:column}
 .msg {padding:4px;margin-bottom:4px;border-radius:4px;width:80%}

open_db.py

@@ -1,7 +1,7 @@
-import sqlite3
+from response_db import ResponseDb
 import pandas as pd
 
-db = sqlite3.connect("response.db")
-df = pd.read_sql('SELECT * from responses', db)
+db = ResponseDb()
+df = pd.DataFrame(list(db.get()))   
 print(df)
-df.to_csv("responses.csv", index=False)
+df.to_csv("responses.csv", index=False)

requirements.txt

@@ -13,4 +13,4 @@ pycocotools
 Pillow
 benepar
 sentence-transformers
-
+pymongo

response_db.py

@@ -1,78 +1,22 @@
-import sqlite3
-from sqlite3 import Connection
+from pymongo import MongoClient
 import datetime
 
 class ResponseDb:
-    DB_PATH = "response.db"
-
     def __init__(self):
-        # establish connection
-        self.con = self.get_connection()
-        cur = self.con.cursor()
-        # create table if it doesnt already exist
-        table_string = "CREATE TABLE IF NOT EXISTS responses (dialogue_id text, task_id text, turn integer, question text, response text, datetime date)"
-        cur.execute(table_string)
-        self.con.commit()
-        cur.close()
-
-    def get_connection(self):
-        """Put the connection in cache to reuse if path does not change between Streamlit reruns.
-        NB : https://stackoverflow.com/questions/48218065/programmingerror-sqlite-objects-created-in-a-thread-can-only-be-used-in-that-sa
-        """
-        return sqlite3.connect(self.DB_PATH, check_same_thread=False)
-
-    #def __del__(self):
-        #self.con.close()
+        # Set up the connection
+        self.client = MongoClient(f"mongodb+srv://{mongodb_username}:{mongodb_pw}@{mongodb_cluster_url}/?retryWrites=true&w=majority")
+        self.db = self.client['vqa-game']
+        self.collection = self.db['vqa-game']
 
     def add(self, dialogue_id, task_id, turn, question, response):
-        cur = self.con.cursor()
         curr_datetime = datetime.datetime.now()
-        cur.execute(
-            "insert into responses values (?, ?, ?, ?, ?, ?)",
-            (dialogue_id, int(task_id), turn, question, response, curr_datetime),
-        )
-        self.con.commit()
-        cur.close()
-
-    def get_id(self, dialogue_id):
-        cur = self.con.cursor()
-        cur.execute(
-            "select * from responses where conv=:id",
-            {"id": dialogue_id},
-        )
-        results = cur.fetchall()
-        cur.close()
-        return results
-
-    def get_id_turn(self, dialogue_id, turn):
-        cur = self.con.cursor()
-        cur.execute(
-            "select * from responses where conv=:id and turn=:turn",
-            {
-                "id": dialogue_id,
-                "turn": turn,
-            },
-        )
-        results = cur.fetchall()
-        cur.close()
-        return results
-
-    def get_all(self):
-        cur = self.con.cursor()
-        cur.execute(
-            "select * from responses",
-        )
-        results = cur.fetchall()
-        cur.close()
-        return results
-
-class StResponseDb(ResponseDb):
-    def get_connection(self):
-        """Put the connection in cache to reuse if path does not change between Streamlit reruns.
-        NB : https://stackoverflow.com/questions/48218065/programmingerror-sqlite-objects-created-in-a-thread-can-only-be-used-in-that-sa
-        """
-        return sqlite3.connect(self.DB_PATH, check_same_thread=False)
-
-if __name__ == "__main__":
-    db = ResponseDb()
-    print(db.get_all())
+        document = {"dialogue_id":dialogue_id,
+                    "task_id":task_id,
+                    "turn":turn,
+                    "question":question,
+                    "response":response,
+                    "datetime":curr_datetime}
+        result = self.collection.insert_one(document)
+
+    def get(self):
+        return self.collection.find()