Update app.py

app.py

@@ -31,7 +31,7 @@ from datasets import load_dataset
 #dataset = load_dataset("csv", data_files="./data.csv")
 
 
-DB_FILE = "./reviews.db"
+DB_FILE = "./reviewsitr.db"
 
 TOKEN = os.environ.get('HF_KEY')
 
@@ -64,7 +64,7 @@ except sqlite3.OperationalError:
         '''
         CREATE TABLE reviews (id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
                               created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL,
-                              name TEXT, rate INTEGER, celsci TEXT)
+                              name TEXT, rate TEXT, celsci TEXT)
         ''')
     db.commit()
     db.close()
@@ -87,22 +87,25 @@ def ccogsphere(name: str, rate: int, celsci: str):
     db = sqlite3.connect(DB_FILE)
     cursor = db.cursor()
       
-    try:
-        celsci2=celsci.split()
-        print("split",celsci2,celsci)
-        celsci2=celsci2[0] + "+" + celsci2[1]
-        celsci2=ecf(celsci2)
-        celsci2=json.dumps(celsci2["videos"])
-    except:
-        celsci2=" No Info Found"
+    #try:
+    celsci2=celsci.split()
+    print("split",celsci2,celsci)
+    celsci2=celsci2[0] + "+" + celsci2[1]
+    celsci2=ecf(celsci2)
+    df=pd.DataFrame.from_dict(celsci2["videos"])
+    celsci2=json.dumps(celsci2["videos"])
+    for index, row in df.iterrows():
+        view = str(row["views"])
+        duration = str(row["duration"])
+        print(view, duration)
+        #celsci=celsci+celsci2
+        cursor.execute("INSERT INTO reviews(name, rate, celsci) VALUES(?,?,?)", [celsci, view, duration])
+        db.commit()
     
-    celsci=celsci+celsci2
-    cursor.execute("INSERT INTO reviews(name, rate, celsci) VALUES(?,?,?)", [name, rate, celsci])
-    db.commit()
     reviews, total_reviews = get_latest_reviews(db)
     db.close()
-    r = requests.post(url='https://ccml-persistent-data2.hf.space/api/predict/', json={"data": [name,celsci]})
-    #demo.load()
+    r = requests.post(url='https://ccml-persistent-data2.hf.space/api/predict/', json={"data": [celsci + " ", celsci2]}) 
+
     return reviews, total_reviews
 
 def run_actr():
@@ -174,7 +177,7 @@ with gr.Blocks(css=css) as demo:
             demo.load(load_data, None, [data, count])
             @name.change(inputs=name, outputs=celsci,_js="window.location.reload()")
             @rate.change(inputs=rate, outputs=name,_js="window.location.reload()")
-            @celsci.change(inputs=celsci, outputs=rate,_js="window.location.reload()")  
+            #@celsci.change(inputs=celsci, outputs=rate,_js="window.location.reload()")  
             
             def secwork(name):
                 #if name=="abc":
@@ -213,7 +216,7 @@ def backup_db_csv():
 #    #return reviews, total_reviews
     
 #scheduler1 = BackgroundScheduler()
-#scheduler1.add_job(func=run_actr, trigger="interval", seconds=3600)
+#scheduler1.add_job(func=run_actr, trigger="interval", seconds=10)
 #scheduler1.start()
 
 scheduler2 = BackgroundScheduler()