Update app.py

app.py

@@ -52,67 +52,66 @@ class Investment_Simulator:
     
     def update_portfolio_history(self):
         now = datetime.now()
-        total_value = compute_portfolio_value()
+        total_value = self.compute_portfolio_value()
         self.portfolio_history.append((now, total_value))
     
     def plot_portfolio_value(self):
         if not self.portfolio_history:
             return None
         df = pd.DataFrame(self.portfolio_history, columns=["Time", "Value"])
-        fig = px.line(df, x="Time", y="Value", title="📈 Portfolio Value Over Time", labels={"Time": "Time", "Value": "Value (£)"})
+        fig = px.line(df, x="Time", y="Value", title="Portfolio Value Over Time", labels={"Time": "Time", "Value": "Value (£)"})
         fig.update_layout(xaxis_rangeslider_visible=True)
         return fig
 
     def process_input(self, user_input):
-        #global self.portfolio, self.history
         user_input = user_input.lower()
     
         try:
             if "buy" in user_input:
                 amount = int(user_input.split("£")[1].split()[0])
                 ticker = user_input.split("of")[1].strip().upper()
-                price = get_price(ticker)
+                price = self.get_price(ticker)
                 if not price:
-                    return f"❌ Couldn't fetch price for {ticker}."
+                    return f"Couldn't fetch price for {ticker}."
                 qty = round(amount / price, 4)
                 if amount > self.portfolio["cash"]:
-                    return f"❌ Insufficient funds (£{self.portfolio['cash']:.2f})."
+                    return f"Insufficient funds (£{self.portfolio['cash']:.2f})."
                 self.portfolio["cash"] -= amount
                 self.portfolio["stocks"][ticker] = self.portfolio["stocks"].get(ticker, 0) + qty
-                self.history.append(f"🟢 Bought £{amount} of {ticker} ({qty} shares at £{price:.2f})")
-                update_portfolio_history()
-                return f"✅ Bought {qty} shares of {ticker} at £{price:.2f}"
+                self.history.append(f"Bought £{amount} of {ticker} ({qty} shares at £{price:.2f})")
+                self.update_portfolio_history()
+                return f"Bought {qty} shares of {ticker} at £{price:.2f}"
     
             elif "sell" in user_input:
                 amount = int(user_input.split("£")[1].split()[0])
                 ticker = user_input.split("of")[1].strip().upper()
-                price = get_price(ticker)
+                price = self.get_price(ticker)
                 if not price:
-                    return f"❌ Couldn't fetch price for {ticker}."
+                    return f"Couldn't fetch price for {ticker}."
                 qty_to_sell = round(amount / price, 4)
                 current_qty = self.portfolio["stocks"].get(ticker, 0)
                 if qty_to_sell > current_qty:
-                    return f"❌ You don't own enough of {ticker}."
+                    return f"You don't own enough of {ticker}."
                 self.portfolio["stocks"][ticker] -= qty_to_sell
                 self.portfolio["cash"] += amount
-                self.history.append(f"🔴 Sold £{amount} of {ticker} ({qty_to_sell} shares at £{price:.2f})")
-                update_portfolio_history()
-                return f"✅ Sold {qty_to_sell} shares of {ticker} at £{price:.2f}"
+                self.history.append(f"Sold £{amount} of {ticker} ({qty_to_sell} shares at £{price:.2f})")
+                self.update_portfolio_history()
+                return f"Sold {qty_to_sell} shares of {ticker} at £{price:.2f}"
     
             elif "portfolio" in user_input:
                 cash = self.portfolio['cash']
                 total_stock_value = 0.0
-                summary = [f"💰 Cash: £{cash:.2f}"]
+                summary = [f"Cash: £{cash:.2f}"]
                 for ticker, qty in self.portfolio["stocks"].items():
-                    live_price = get_price(ticker)
+                    live_price = self.get_price(ticker)
                     if live_price:
                         value = qty * live_price
                         total_stock_value += value
-                        summary.append(f"📈 {ticker}: {qty:.4f} shares (£{value:.2f})")
+                        summary.append(f"{ticker}: {qty:.4f} shares (£{value:.2f})")
                     else:
-                        summary.append(f"📈 {ticker}: {qty:.4f} shares (price unavailable)")
+                        summary.append(f"{ticker}: {qty:.4f} shares (price unavailable)")
                 total_value = cash + total_stock_value
-                summary.append(f"\n💼 Total Portfolio Value: £{total_value:.2f}")
+                summary.append(f"\nTotal Portfolio Value: £{total_value:.2f}")
                 return "\n".join(summary)
     
             elif "history" in user_input:
@@ -123,21 +122,21 @@ class Investment_Simulator:
                 self.portfolio["stocks"].clear()
                 self.history.clear()
                 self.portfolio_history.clear()
-                update_portfolio_history()
-                return "🔁 Portfolio reset."
+                self.update_portfolio_history()
+                return "Portfolio reset."
     
             else:
-                return "❓ Try commands like:\n• Buy £100 of AAPL\n• Sell £50 of TSLA\n• Show portfolio"
+                return "Try commands like:\n• Buy £100 of AAPL\n• Sell £50 of TSLA\n• Show portfolio"
     
         except Exception as e:
-            return f"⚠️ Error: {e}"
+            return f"Error: {e}"
 
     def run_all(self, text):
-        response = process_input()
-        fig = plot_portfolio_value()
+        response = self.process_input()
+        fig = self.plot_portfolio_value()
         return response, fig
     
-    update_portfolio_history()
+    self.update_portfolio_history()
 
 
 class News: