Update app.py

app.py

@@ -65,7 +65,7 @@ except sqlite3.OperationalError:
         '''
         CREATE TABLE reviews (id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
                               created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL,
-                              name TEXT, view TEXT, duration TEXT)
+                              name TEXT, view TEXT, duration TEXT, x TEXT)
         ''')
     db.commit()
     db.close()
@@ -75,7 +75,7 @@ except sqlite3.OperationalError:
 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]
-    reviews = pd.DataFrame(reviews, columns=["id", "date_created", "name", "view", "duration"])
+    reviews = pd.DataFrame(reviews, columns=["id", "date_created", "name", "view", "duration", "X"])
     return reviews, total_reviews
 
 def get_latest_reviews2(db: sqlite3.Connection):
@@ -91,9 +91,6 @@ def ccogsphere(name: str, rate: str, celsci: str):
     celsci2=celsci.split()
     celsci2=celsci2[0] + "+" + celsci2[1]
     celscix=celsci2[0] + celsci2[1]
-    celscix2=gcf(celscix)
-    if celscix={"message":"Cannot find information about Twitter Screen Name 'donald trump'"}:
-        print ("no")
 
     if celsci=="Celscis List":
         df = pd.DataFrame({
@@ -110,10 +107,14 @@ def ccogsphere(name: str, rate: str, celsci: str):
         else:
             nam="No name, write your name followed by a message"
             mess="No messages, write your name followed by a message"
-       
-        cursor.execute("INSERT INTO reviews(name, view, duration) VALUES(?,?,?)", [nam, " wrote message: " , mess])
+        celscix2="End"
+        cursor.execute("INSERT INTO reviews(name, view, duration,celscix2) VALUES(?,?,?,?)", [nam, " wrote message: " , mess, celscix2])
         db.commit()
     else:
+        celscix2=gcf(celscix)
+        #if celscix=={"message":"Cannot find information about Twitter Screen Name 'donald trump'"}:
+        #    print ("no")
+        #else:
         celsci2=ecf(celsci2)
         df=pd.DataFrame.from_dict(celsci2["videos"])
         celsci2=json.dumps(celsci2["videos"])
@@ -122,12 +123,12 @@ def ccogsphere(name: str, rate: str, celsci: str):
             duration = str(row["duration"])
             print(view, duration)
             #celsci=celsci+celsci2
-            cursor.execute("INSERT INTO reviews(name, view, duration) VALUES(?,?,?)", [celsci+str(index+1), view, duration])
+            cursor.execute("INSERT INTO reviews(name, view, duration,celscix2) VALUES(?,?,?,?)", [celsci+str(index+1), view, duration, celscix2])
             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": [celsci + " ", celsci2]}) 
+    r = requests.post(url='https://ccml-persistent-data2.hf.space/api/predict/', json={"data": [celsci + " ", celsci2+celscix2]}) 
 
     return reviews, total_reviews