Update app.py

app.py

@@ -86,7 +86,7 @@ def ccogsphere(name: str, rate: int, celsci: str):
     db.close()
     r = requests.post(url='https://ccml-persistent-data2.hf.space/api/predict/', json={"data": [name,celsci]})
     #demo.load()
-    inp=celsci.split(" ")
+    inp=celsci.split()
     inp=inp[0] + "+" + inp[1]
     result=ecf(inp)
     df=pd.DataFrame.from_dict(result["videos"])
@@ -127,11 +127,25 @@ def run_ecs(inp):
     return df
 
   
-def load_data():
+def load_data(celsci: str):
     db = sqlite3.connect(DB_FILE)
     reviews, total_reviews = get_latest_reviews(db)
     db.close()
-    return reviews, total_reviews
+    if celsci!="":
+        inp=celsci.split()
+        inp=inp[0] + "+" + inp[1]
+        result=ecf(inp)
+        df=pd.DataFrame.from_dict(result["videos"])
+    else:
+        # Creating a sample dataframe
+        df = pd.DataFrame({
+            "A" : [14, 4, 5, 4, 1], 
+            "B" : [5, 2, 54, 3, 2], 
+            "C" : [20, 20, 7, 3, 8], 
+            "D" : [14, 3, 6, 2, 6], 
+            "E" : [23, 45, 64, 32, 23]
+        }) 
+    return df, reviews, total_reviews
 def load_data2():
     #result=run_ecs(celscie)
     #df = pd.DataFrame.from_dict(result["videos"])
@@ -167,7 +181,7 @@ with gr.Blocks() as demo:
             #run_actr()
             submit = gr.Button(value=".")
             submit.click(ccogsphere, [name, rate, celsci], [data2, data, count])
-            demo.load(load_data, None, [data, count])
+            demo.load(load_data, celsci, [data2, 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()")  
@@ -175,7 +189,7 @@ with gr.Blocks() as demo:
             def secwork(name):
                 #if name=="abc":
                 #run_code()
-                load_data()
+                load_data("")
                 #return "Hello " + name + "!"
     with gr.Row():
         with gr.Column():