Upload folder using huggingface_hub

app.py

@@ -75,12 +75,14 @@ async def get_wallet_base_balances(addresses):
         df.drop(columns=['address','decimals'],inplace=True)
 
         df = df[df['symbol'].isin(['VICE', 'ETH', 'USDT', 'USDC'])]
+        df_lp_only = df[df['address'] == '0x45e7035499bc860dA0b5666A93B97AFc51F3cD3F']
+        df_lp_only = df_lp_only[['symbol','tokenBalance','value']]
         df= df[['symbol','tokenBalance','value']]
         df = df.rename(columns={'tokenBalance':'amount','value':'usd_value'})
         
         df = df.groupby("symbol", as_index=False).sum()
 
-        return df, df['usd_value'].sum(), eth_price, base_price
+        return df, df['usd_value'].sum(), eth_price, base_price, df_lp_only
 
 async def fetch_lp():
     url = "https://public.zapper.xyz/graphql"
@@ -213,8 +215,12 @@ async def fetch_lp():
     
 
 async def get_all_dfs():
-    #starting_df = pd.DataFrame([{"symbol":"AIQ","amount":3000000.00,"usd_value":3000000.00*0.05},{"symbol":"ETH","amount":55.5309,"usd_value":55.5309*4486.62}])
-    df, total, eth_price, base_price = await get_wallet_base_balances(base_addresses)
+    starting_df = pd.DataFrame({
+    "symbol": ["ETH", "VICE"],
+    "amount": [23.77, 11066328.78],
+    "usd_value": [111264.43, 448971.41]
+    })
+    df, total, eth_price, base_price, df_lp_only = await get_wallet_base_balances(base_addresses)
     symbols = {"VICE":base_price,"WETH":eth_price}
     lp_df = await fetch_lp()
     if not lp_df.empty:
@@ -224,18 +230,20 @@ async def get_all_dfs():
     for col in ['amount','usd_value']:
         df[col] = pd.to_numeric(df[col], errors='coerce')
         df[col] = df[col].apply(lambda x: f"{x:,.2f}")
-    return df,lp_df
+    return starting_df,df,lp_df,df_lp_only
 
 with gr.Blocks() as demo:
     gr.Markdown("## VICE Balances & PnL")
 
+    df0_out = gr.Dataframe(label="Starting Balances")
     df1_out = gr.Dataframe(label="Current Balances")
-    df2_out = gr.Dataframe(label="LP Balances")
-
+    df3_out = gr.Dataframe(label="LP Wallet Balances")
+    df2_out = gr.Dataframe(label="LP Position Balances")
+    
     demo.load(
     fn= get_all_dfs, 
     inputs = [],
-    outputs=[df1_out,df2_out]
+    outputs=[df0_out,df1_out,df2_out,df3_out]
     )