Add trades tables

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 import pandas as pd
 from pymongo import MongoClient
 import plotly.express as px
+import plotly.graph_objects as go
 
 DB_USER = os.getenv("DB_USER")
 DB_PASSWORD = os.getenv("DB_PASSWORD")
@@ -29,15 +30,19 @@ def fetch_orders_for_strategy(strategy):
     orders = orders.dropna(axis=0, subset=['open_price', 'close_price'])
     orders['open_price'] =  orders['open_price'].astype(str).astype(float)
     orders['close_price'] =  orders['close_price'].astype(str).astype(float)
+    # orders['created_ts'] = pd.to_datetime(orders['created_ts'])
     orders['created_ts'] = pd.to_datetime(orders['created_ts'])
     orders['closed_ts'] = pd.to_datetime(orders['closed_ts'])
     orders['profit'] = orders['close_price'] - orders['open_price']
+    orders['profit'] = orders['profit'].round(2)
     orders['created_date'] = orders['created_ts'].dt.date
+    orders['ticker_open_price'] = orders['ticker_open_price'].astype(int)
+    orders['ticker_close_price'] = orders['ticker_close_price'].astype(int)
     orders = orders.sort_values(by='created_ts', ascending=False)
     return orders
 
 def create_latest_profit_plot(strategy, profits):
-    if "Latest" != profits:
+    if profits not in ["Profits", "Trades"]:
         return gr.update(visible=False)
     orders = fetch_orders_for_strategy(strategy)
     latest_day = orders[orders['strategy_id'] == strategy].sort_values(by='created_ts', ascending=False)['created_date'].unique()[0]
@@ -48,7 +53,28 @@ def create_latest_profit_plot(strategy, profits):
     plot = px.line(cumsum, x="created_ts", y="profit",
                    title=f"Profits for day for {strategy}")
     plot.update_layout(legend=dict(x=0.5, y=0.99),  title_x=0.5, legend_title_text="")
-    return gr.update(value=plot, visible=True)
+
+    table = go.Figure(data=[go.Table(
+        header=dict(values=list(['Strategy', 'Created', 'Closed', 'Ticker Open', 'Ticker Close', 'Market Type', 'Symbol', 'Open', 'Close', 'Qty', 'Profit']),
+                    align='left'),
+        cells=dict(values=[orders.strategy_id,
+                            orders.created_ts,
+                            orders.closed_ts,
+                            orders.ticker_open_price,
+                            orders.ticker_close_price,
+                            orders.market_type,
+                            orders.tradingsymbol,
+                            orders.open_price,
+                            orders.close_price,
+                            orders.quantity,
+                            orders.profit])
+                )
+    ])
+
+    if profits == 'Profits':
+        return gr.update(value=plot, visible=True)
+    if profits == 'Trades':
+        return gr.update(value=table, visible=True)
 
 
 with gr.Blocks() as demo:
@@ -60,15 +86,19 @@ with gr.Blocks() as demo:
         with gr.Column():
             with gr.Box():
                 gr.Markdown("## Select graphs to display")
-                profits = gr.Radio(choices=["Latest", "All"], label="", value="Latest")
+                type = gr.Radio(choices=["Profits", "Trades"], label="", value="Profits")
     with gr.Row():
         fetch = gr.Button(value="Fetch")
     with gr.Row():
         with gr.Column():
             profits_plot = gr.Plot(visible=False)
+    with gr.Row():
+        with gr.Column():
+            trades_plot = gr.Plot(visible=False)
+
 
-    fetch.click(create_latest_profit_plot, inputs=[strategy, profits], outputs=profits_plot)
-    # fetch.click(create_star_plot, inputs=[libraries, stars], outputs=star_plot)
+    fetch.click(create_latest_profit_plot, inputs=[strategy, type], outputs=profits_plot)
+    # fetch.click(create_latest_profit_plot, inputs=[strategy, type], outputs=trades_plot)
     # fetch.click(create_issue_plot, inputs=[libraries, issues], outputs=issue_plot)