Update structure

app.py

@@ -8,7 +8,6 @@ import datetime
 from apscheduler.schedulers.background import BackgroundScheduler
 
 
-
 DB_FILE = "./reviews.db"
 
 TOKEN = os.environ.get('HUB_TOKEN')
@@ -24,6 +23,8 @@ repo.git_pull()
 shutil.copyfile("./data/reviews.db", DB_FILE)
 
 
+# Create table if it doesn't already exist
+
 db = sqlite3.connect(DB_FILE)
 try:
     db.execute("SELECT * FROM reviews").fetchall()
@@ -39,13 +40,11 @@ except sqlite3.OperationalError:
     db.close()
 
 
-def backup_db():
-    shutil.copyfile(DB_FILE, "./data/reviews.db")
-    db = sqlite3.connect(DB_FILE)
-    reviews = db.execute("SELECT * FROM reviews").fetchall()
-    pd.DataFrame(reviews).to_csv("./data/reviews.csv", index=False)
-    print("updating db")
-    repo.push_to_hub(blocking=False, commit_message=f"Updating data at {datetime.datetime.now()}")
+def get_latest_reviews(db: sqlite3.Connection):
+    reviews = db.execute("SELECT * FROM reviews ORDER BY id DESC limit 10").fetchall()
+    total_reviews = db.execute("Select COUNT(id) from reviews").fetchone()[0]
+    reviews = pd.DataFrame(reviews, columns=["id", "date_created", "name", "review", "comments"])
+    return reviews, total_reviews
 
 
 def add_review(name: str, review: int, comments: str):
@@ -53,23 +52,16 @@ def add_review(name: str, review: int, comments: str):
     cursor = db.cursor()
     cursor.execute("INSERT INTO reviews(name, review, comments) VALUES(?,?,?)", [name, review, comments])
     db.commit()
-    reviews = db.execute("SELECT * FROM reviews ORDER BY id DESC limit 10").fetchall()
-    total_reviews = db.execute("Select COUNT(id) from reviews").fetchone()[0]
+    reviews, total_reviews = get_latest_reviews(db)
     db.close()
-    return pd.DataFrame(reviews), total_reviews
+    return reviews, total_reviews
 
 
 def load_data():
     db = sqlite3.connect(DB_FILE)
-    reviews = db.execute("SELECT * FROM reviews ORDER BY created_at DESC limit 10").fetchall()
-    total_reviews = db.execute("Select COUNT(id) from reviews").fetchone()[0]
+    reviews, total_reviews = get_latest_reviews(db)
     db.close()
-    return pd.DataFrame(reviews), total_reviews
-
-
-scheduler = BackgroundScheduler()
-scheduler.add_job(func=backup_db, trigger="interval", seconds=60)
-scheduler.start()
+    return reviews, total_reviews
 
 
 with gr.Blocks() as demo:
@@ -80,9 +72,24 @@ with gr.Blocks() as demo:
             comments = gr.Textbox(label="Comments", lines=10, placeholder="Do you have any feedback on gradio?")
             submit = gr.Button(value="Submit Feedback")
         with gr.Column():
-            data = gr.DataFrame(label="Most recently created 10 rows", headers=["id", "date_created", "name", "review", "comments"])
+            data = gr.Dataframe(label="Most recently created 10 rows")
             count = gr.Number(label="Total number of reviews")
     submit.click(add_review, [name, review, comments], [data, count])
     demo.load(load_data, None, [data, count])
 
+
+def backup_db():
+    shutil.copyfile(DB_FILE, "./data/reviews.db")
+    db = sqlite3.connect(DB_FILE)
+    reviews = db.execute("SELECT * FROM reviews").fetchall()
+    pd.DataFrame(reviews).to_csv("./data/reviews.csv", index=False)
+    print("updating db")
+    repo.push_to_hub(blocking=False, commit_message=f"Updating data at {datetime.datetime.now()}")
+
+
+scheduler = BackgroundScheduler()
+scheduler.add_job(func=backup_db, trigger="interval", seconds=60)
+scheduler.start()
+
+
 demo.launch()