Update utils/ui_helper.py

utils/ui_helper.py

@@ -36,6 +36,7 @@ def main_algo_trader():
         ],
     )
     top_n = st.sidebar.number_input("Top n stocks", min_value=1, value=3)
+    height_of_graph = st.sidebar.number_input("Height of the plot", min_value=500, value=750)
 
     # Process inputs
     tickers_list = [ticker.strip() for ticker in tickers.split(",")]
@@ -64,6 +65,14 @@ def main_algo_trader():
             ]
             df.index = pd.to_datetime(df.index, unit="ms")
 
+            # Create download file
+            @st.cache_data
+            def convert_df(df):
+                # IMPORTANT: Cache the conversion to prevent computation on every rerun
+                return df.to_csv().encode('utf-8')
+            
+            csv = convert_df(df)
+
             # Create plot
             fig = go.Figure()
             fig.add_trace(
@@ -140,7 +149,15 @@ def main_algo_trader():
                 legend=dict(
                     orientation="h", x=0.5, y=-0.4, xanchor="center", yanchor="bottom"
                 ),
-                height=600,
+                height=height_of_graph,
             )
 
-            st.plotly_chart(fig, use_container_width=True)
+            st.plotly_chart(fig, use_container_width=True)
+
+            # Download
+            st.download_button(
+                label="Download data as CSV",
+                data=csv,
+                file_name=f'history_{end_date}.csv',
+                mime='text/csv',
+            )