Add application file

app.py

@@ -0,0 +1,76 @@
+
+import os
+import gradio as gr
+import pandas as pd
+from pymongo import MongoClient
+import plotly.express as px
+
+DB_USER = os.getenv("DB_USER")
+DB_PASSWORD = os.getenv("DB_PASSWORD")
+DB_HOST = os.getenv("DB_HOST")
+client = MongoClient(host=f"mongodb+srv://{DB_USER}:{DB_PASSWORD}@{DB_HOST}/?retryWrites=true&w=majority")
+
+#select database
+db = client['algotrading']
+#select the collection within the database
+
+def fetch_all_strategy():
+    global db
+    orders_t = db.orders
+    strategies = pd.DataFrame(list(orders_t.find({}, {"strategy_id":1})))
+    return list(strategies['strategy_id'].unique())
+
+
+def fetch_orders_for_strategy(strategy):
+    global db
+    orders_t = db.orders
+    #convert entire collection to Pandas dataframe
+    orders = pd.DataFrame(list(orders_t.find({"strategy_id": 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['closed_ts'] = pd.to_datetime(orders['closed_ts'])
+    orders['profit'] = orders['close_price'] - orders['open_price']
+    orders['created_date'] = orders['created_ts'].dt.date
+    orders = orders.sort_values(by='created_ts', ascending=False)
+    return orders
+
+def create_latest_profit_plot(strategy, profits):
+    if "Latest" != profits:
+        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]
+    cumsum = orders[(orders['created_date'] == latest_day) & (orders['strategy_id'] == strategy)] \
+        .sort_values(by='created_ts', ascending=True) \
+        .groupby(['strategy_id', 'created_ts'])['profit'].sum() \
+        .groupby(level=0).cumsum().reset_index()
+    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)
+
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            with gr.Box():
+                gr.Markdown("## Select strategies to display")
+                strategy = gr.Radio(choices=fetch_all_strategy(), label="")
+        with gr.Column():
+            with gr.Box():
+                gr.Markdown("## Select graphs to display")
+                profits = gr.Radio(choices=["Latest", "All"], label="", value="Latest")
+    with gr.Row():
+        fetch = gr.Button(value="Fetch")
+    with gr.Row():
+        with gr.Column():
+            profits_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_issue_plot, inputs=[libraries, issues], outputs=issue_plot)
+
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+pymongo
+pandas
+plotly