Update app.py

app.py

@@ -29,7 +29,7 @@ import time
 #hf_hub_download(repo_id="CogSphere/aCogSphere", filename="./reviews.csv")
 
 #from huggingface_hub import login
-#from datasets import load_dataset
+from datasets import load_dataset
 
 #dataset = load_dataset("csv", data_files="./data.csv")
 
@@ -81,11 +81,11 @@ def get_latest_reviews(): #db: sqlite3.Connection):
     reviews = pd.DataFrame(reviews, columns=["id", "date_created", "name", "view", "duration"])
     return reviews, total_reviews
 
-def get_latest_reviews2(db: sqlite3.Connection):
-    reviews2 = db.execute("SELECT * FROM reviews2 ORDER BY id DESC limit 100").fetchall()
-    total_reviews2 = db.execute("Select COUNT(id) from reviews2").fetchone()[0]
-    reviews2 = pd.DataFrame(reviews2, columns=["id","title", "link","channel", "description", "views", "uploaded", "duration", "durationString"])
-    return reviews2, total_reviews2
+#def get_latest_reviews2(db: sqlite3.Connection):
+#    reviews2 = db.execute("SELECT * FROM reviews2 ORDER BY id DESC limit 100").fetchall()
+#    total_reviews2 = db.execute("Select COUNT(id) from reviews2").fetchone()[0]
+#    reviews2 = pd.DataFrame(reviews2, columns=["id","title", "link","channel", "description", "views", "uploaded", "duration", "durationString"])
+#    return reviews2, total_reviews2
     
 def ccogsphere(name: str, rate: int, celsci: str):
     #db = sqlite3.connect(DB_FILE)
@@ -156,11 +156,11 @@ def load_data():
     reviews, total_reviews = get_latest_reviews(db)
     #db.close()
     return reviews, total_reviews
-def load_data2():
-    db = sqlite3.connect(DB_FILE)
-    reviews2, total_reviews2 = get_latest_reviews2(db)
-    db.close()
-    return reviews2, total_reviews2
+#def load_data2():
+#    db = sqlite3.connect(DB_FILE)
+#    reviews2, total_reviews2 = get_latest_reviews2(db)
+#    db.close()
+#    return reviews2, total_reviews2
     
 css="footer {visibility: hidden}"
 # Applying style to highlight the maximum value in each row
@@ -189,17 +189,17 @@ with gr.Blocks(css=css) as demo:
                 load_data()
                 #return "Hello " + name + "!"
 def backup_db():
-    shutil.copyfile(DB_FILE, "./reviews.db")
-    db = sqlite3.connect(DB_FILE)
-    reviews = db.execute("SELECT * FROM reviews").fetchall()
+    #shutil.copyfile(DB_FILE, "./reviews.db")
+    #db = sqlite3.connect(DB_FILE)
+    #reviews = db.execute("SELECT * FROM reviews").fetchall()
     pd.DataFrame(reviews).to_csv("./reviews.csv", index=False)
     print("updating db")
     repo.push_to_hub(blocking=False, commit_message=f"Updating data at {datetime.datetime.now()}")
     
 def backup_db_csv():
-    shutil.copyfile(DB_FILE, "./reviews2.db")
-    db = sqlite3.connect(DB_FILE)
-    reviews = db.execute("SELECT * FROM reviews").fetchall()
+    #shutil.copyfile(DB_FILE, "./reviews2.db")
+    #db = sqlite3.connect(DB_FILE)
+    #reviews = db.execute("SELECT * FROM reviews").fetchall()
     pd.DataFrame(reviews).to_csv("./reviews2.csv", index=False)
     print("updating db csv")
     dataset = load_dataset("csv", data_files="./reviews2.csv")