Upload folder using huggingface_hub

app.py

@@ -223,6 +223,10 @@ async def process_data():
 
   return(starting_df,final_df.tail(1),jup_df.tail(1),df_with_totals.tail(1))
 
+client = MongoClient('mongodb+srv://djamaal:FHmV2N733lzrLkWf@test-cluster.mys4n.mongodb.net/')
+db = client["billy_balances"]
+collection = db["balance_data"]
+
 async def display_data():
     df0, df1, df2, df3 = await process_data()
 
@@ -232,11 +236,47 @@ async def display_data():
         return df, total_value
 
     df0, total0 = summarize(df0)
+    df1, total1 = summarize(df1)
+    df2, total2 = summarize(df2)
     df3, total3 = summarize(df3)
 
     pnl = float(total3) - float(total0)
 
-    return summarize(df0) + summarize(df1) + summarize(df2) + summarize(df3) + (f"PNL : {pnl:,.2f}"," ")
+    # Save to MongoDB
+    record = {
+        "timestamp": datetime.datetime.utcnow(),
+        "df0": df0.to_dict(),
+        "total0": total0,
+        "df1": df1.to_dict(),
+        "total1": total1,
+        "df2": df2.to_dict(),
+        "total2": total2,
+        "df3": df3.to_dict(),
+        "total3": total3,
+        "pnl": pnl
+    }
+    collection.insert_one(record)
+
+    return df0, total0, df1, total1, df2, total2, df3, total3, f"PNL : {pnl:,.2f}"
+
+def load_from_mongo():
+    latest = collection.find_one(sort=[("timestamp", -1)])
+    if latest:
+        df0 = pd.DataFrame(latest["df0"])
+        df1 = pd.DataFrame(latest["df1"])
+        df2 = pd.DataFrame(latest["df2"])
+        df3 = pd.DataFrame(latest["df3"])
+
+        return (
+            df0, latest["total0"],
+            df1, latest["total1"],
+            df2, latest["total2"],
+            df3, latest["total3"],
+            f"PNL : {latest['pnl']:,.2f}"
+        )
+    else:
+        empty_df = pd.DataFrame()
+        return empty_df, 0, empty_df, 0, empty_df, 0, empty_df, 0, "PNL : 0.00"
 
 with gr.Blocks() as demo:
     gr.Markdown("## BILLY Balances and PnL")
@@ -256,13 +296,30 @@ with gr.Blocks() as demo:
 
     pnl_out = gr.Textbox(label="PNL (AUM Model)")
 
-    # Set up the function to run on app load
+    # Load from MongoDB on app load (sync function)
     demo.load(
-        fn=display_data,
+        fn=load_from_mongo,
         inputs=[],
-        outputs=[df0_out, txt0_out, df1_out, txt1_out, df2_out, txt2_out, df3_out, txt3_out, pnl_out]
+        outputs=[
+            df0_out, txt0_out,
+            df1_out, txt1_out,
+            df2_out, txt2_out,
+            df3_out, txt3_out,
+            pnl_out
+        ]
     )
 
-    btn.click(display_data, outputs=[df0_out, txt0_out, df1_out, txt1_out, df2_out, txt2_out, df3_out,txt3_out,pnl_out])
+    # Async button click — compute and store new data
+    btn.click(
+        fn=display_data,
+        inputs=[],
+        outputs=[
+            df0_out, txt0_out,
+            df1_out, txt1_out,
+            df2_out, txt2_out,
+            df3_out, txt3_out,
+            pnl_out
+        ]
+    )
 
-demo.launch(debug=True,share=True)
+demo.launch(debug=True, share=True)