Update app.py

app.py

@@ -22,10 +22,10 @@ def create_portfolio_graph(amount):
 
     return chart
 
-def on_submit(button):
-    amount = button.inputs[0]
-    plot = button.outputs[0]
-    plot.update(create_portfolio_graph(amount))
+def on_submit():
+    # Callback function to handle button click
+    amount = total_funds_input.value
+    plot.change(create_portfolio_graph, inputs=[total_funds_input], outputs=[plot])
 
 with gr.Blocks() as demo:
     title = gr.Markdown(
@@ -34,8 +34,9 @@ with gr.Blocks() as demo:
         """
     )
     total_funds_input = gr.Number(value=0)
+    button = gr.Button("Submit", click=on_submit)
     plot = gr.Plot(label="Portfolio Composition Chart")
-    button = gr.Button("Submit", onclick=on_submit, args=[total_funds_input])
+    
     demo.load(create_portfolio_graph, inputs=[total_funds_input, button], outputs=[title, plot])
 
 if __name__ == "__main__":
@@ -62,5 +63,6 @@ if __name__ == "__main__":
 
 
 
+