Update app.py

app.py

@@ -77,8 +77,8 @@ from datasets import load_dataset
 def get_latest_reviews(): #db: sqlite3.Connection):
     #reviews = db.execute("SELECT * FROM reviews ORDER BY id DESC limit 100").fetchall()
     #total_reviews = db.execute("Select COUNT(id) from reviews").fetchone()[0]
-    total_reviews="N"
-    reviews = pd.DataFrame(reviews, columns=["id", "date_created", "name", "view", "duration"])
+    reviews = reviews.DataFrame(reviews, columns=["id", "date_created", "name", "view", "duration"])
+    total_reviews=df.count()[0]
     return reviews, total_reviews
 
 #def get_latest_reviews2(db: sqlite3.Connection):
@@ -104,9 +104,10 @@ def ccogsphere(name: str, rate: int, celsci: str):
         print(view, duration)
         #celsci=celsci+celsci2
         cursor.execute("INSERT INTO reviews(name, view, duration) VALUES(?,?,?)", [celsci+" Video #"+str(index+1), view, duration])
-        db.commit()
-    
-    reviews, total_reviews = get_latest_reviews(db)
+        #db.commit()
+    reviews=df
+    total_reviews=df.count()[0]
+    #reviews, total_reviews = get_latest_reviews(db)
     #db.close()
     r = requests.post(url='https://ccml-persistent-data2.hf.space/api/predict/', json={"data": [celsci + " ", celsci2]}) 
 
@@ -137,7 +138,7 @@ def run_ecs(inp):
     except: # sqlite3.OperationalError:
         print ("db error")
     
-    df=df.drop(df.columns[4], axis=1)
+    #df=df.drop(df.columns[4], axis=1)
 
     #db = sqlite3.connect(DB_FILE)
     #cursor = db.cursor()
@@ -145,7 +146,7 @@ def run_ecs(inp):
     #df.to_sql('reviews2', db, if_exists='replace', index=False)
 
     #db.commit()
-    reviews2, total_reviews2 = get_latest_reviews(db)
+    #reviews2, total_reviews2 = get_latest_reviews(db)
     #db.close()
     #print ("print000", total_reviews2,reviews2)
     return reviews2, total_reviews2