Update app.py

app.py

@@ -2,10 +2,14 @@ import gradio as gr
 import altair as alt
 import pandas as pd
 
-def create_portfolio_graph():
+def create_portfolio_graph(amount):
+    # Calculate positions based on the provided amount
+    long_position = 0.9 * amount
+    short_position = 0.1 * amount
+
     # Data for the bars
     positions = ['Long', 'Short']
-    percentages = [0.9, 0.1]  # 90% long, 10% short
+    percentages = [long_position, short_position]
 
     data = pd.DataFrame({'Position': positions, 'Percentage': percentages})
 
@@ -24,9 +28,13 @@ with gr.Blocks() as demo:
         # This is the Current Recommended Portfolio Allocation
         """
     )
+    total_funds_input = gr.Textbox("Type your total funds", default="100000", numeric=True)
+    submit_button = gr.Button("Submit")
+
     plot = gr.Plot(label="Portfolio Composition Chart")
-    plot.change(create_portfolio_graph, inputs=[], outputs=[plot])
-    demo.load(create_portfolio_graph, inputs=[], outputs=[title, plot])
+    plot.change(create_portfolio_graph, inputs=[total_funds_input, submit_button], outputs=[plot])
+    
+    demo.load(create_portfolio_graph, inputs=[total_funds_input, submit_button], outputs=[title, plot])
 
 if __name__ == "__main__":
     demo.launch()
@@ -49,3 +57,4 @@ if __name__ == "__main__":
 
 
 
+