Update app.py

app.py

@@ -132,7 +132,7 @@ class Investment_Simulator:
             return f"Error: {e}"
 
     def run_all(self, text):
-        response = self.process_input()
+        response = self.process_input(text)
         fig = self.plot_portfolio_value()
         return response, fig
     
@@ -203,6 +203,8 @@ custom_css = """
 }
 """
 
+news = News()
+
 with gr.Blocks(css=custom_css) as demo:
     active_section = gr.State("chat")
     sidebar_open = gr.State(True)
@@ -246,14 +248,17 @@ with gr.Blocks(css=custom_css) as demo:
             with Zeno_Investments:
                 gr.Markdown("## Live Investment Simulator (Practice £500) + Portfolio Tracker")
                 
-                simulator = Investment_Simulator(portfolio=None, history=None, portfolio_history=None)
+                simulator = Investment_Simulator()
                 
                 invest_input = gr.Textbox(label="Enter a command", placeholder="Try: Buy £100 of AAPL")
                 invest_output = gr.Textbox(label="Bot Response")
                 invest_chart = gr.Plot(label="Portfolio Value Over Time")
                 
-                invest_input.submit(run_all, inputs=invest_input, outputs=[invest_output, invest_chart])
-
+                invest_input.submit(
+                    simulator.run_all,
+                    inputs=invest_input,
+                    outputs=[invest_output, invest_chart]
+                )
             
             Zeno_News = gr.Column(visible=False, elem_id="Stock and Market News")
             with Zeno_News:
@@ -266,11 +271,11 @@ with gr.Blocks(css=custom_css) as demo:
 
                 fetch_button = gr.Button("Get News")
                 fetch_button.click(
-                    fn=news.get_news,
-                    inputs=ticker_input,
-                    outputs=news_output
+                fn=news.get_news,
+                inputs=ticker_input,
+                outputs=news_output
                 )
-
+                
                 
     def handle_first_input(msg):
         return gr.update(visible=False), gr.update(visible=True), msg, []