Update app.py

app.py

@@ -5,8 +5,9 @@ import matplotlib.pyplot as plt
 import gradio as gr
 import io
 
+print("Starting app...")
+
 def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
-    print("Fetching data...")
     try:
         df = yf.download(ticker, start=start_date, end=end_date, progress=False)
         if df.empty:
@@ -15,10 +16,8 @@ def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
         return None, f"Error fetching data: {str(e)}", None
 
     df = df[['Close']]
-
     df['SMA_50'] = df['Close'].rolling(window=50).mean()
     df['SMA_150'] = df['Close'].rolling(window=150).mean()
-
     df['Signal'] = 0
     df['Signal'][df['SMA_50'] > df['SMA_150']] = 1
     df['Signal'][df['SMA_50'] < df['SMA_150']] = -1
@@ -28,7 +27,6 @@ def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
     shares = 0
     portfolio_values = []
 
-    print("Starting simulation...")
     for index, row in df.iterrows():
         if pd.isna(row['Close']):
             continue
@@ -38,7 +36,6 @@ def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
         elif row['Position'] == -1 and shares > 0:
             cash = shares * row['Close']
             shares = 0
-
         portfolio_value = cash + (shares * row['Close'])
         portfolio_values.append(portfolio_value)