Trireme commited on
Commit
fafdcb8
·
1 Parent(s): a211984

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +14 -10
app.py CHANGED
@@ -230,6 +230,10 @@ client = MongoClient('mongodb+srv://djamaal:FHmV2N733lzrLkWf@test-cluster.mys4n.
230
  db = client["billy_balances"]
231
  collection = db["balance_data"]
232
 
 
 
 
 
233
  async def display_data():
234
  df0, df1, df2, df3 = await process_data()
235
 
@@ -247,16 +251,16 @@ async def display_data():
247
 
248
  # Save to MongoDB
249
  record = {
250
- "timestamp": datetime.datetime.utcnow(),
251
- "df0": df0.to_dict(),
252
- "total0": total0,
253
- "df1": df1.to_dict(),
254
- "total1": total1,
255
- "df2": df2.to_dict(),
256
- "total2": total2,
257
- "df3": df3.to_dict(),
258
- "total3": total3,
259
- "pnl": pnl
260
  }
261
  collection.insert_one(record)
262
 
 
230
  db = client["billy_balances"]
231
  collection = db["balance_data"]
232
 
233
+ def df_to_mongo_safe_dict(df):
234
+ return {str(k): v for k, v in df.to_dict(orient="index").items()}
235
+
236
+
237
  async def display_data():
238
  df0, df1, df2, df3 = await process_data()
239
 
 
251
 
252
  # Save to MongoDB
253
  record = {
254
+ "timestamp": datetime.datetime.utcnow(),
255
+ "df0": df_to_mongo_safe_dict(df0),
256
+ "total0": float(total0),
257
+ "df1": df_to_mongo_safe_dict(df1),
258
+ "total1": float(total1),
259
+ "df2": df_to_mongo_safe_dict(df2),
260
+ "total2": float(total2),
261
+ "df3": df_to_mongo_safe_dict(df3),
262
+ "total3": float(total3),
263
+ "pnl": float(pnl)
264
  }
265
  collection.insert_one(record)
266