Update app.py

app.py

@@ -1,10 +1,10 @@
 import pandas as pd
 import yfinance as yf
-import numpy as np
 import matplotlib.pyplot as plt
 import io
 import gradio as gr
 
+
 def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
     try:
         df = yf.download(ticker, start=start_date, end=end_date, progress=False)
@@ -26,7 +26,7 @@ def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
     shares = 0
     portfolio_values = []
 
-    for index, row in df.iterrows():
+    for _, row in df.iterrows():
         if pd.isna(row['Close']):
             continue
         if row['Position'] == 1 and cash > 0:
@@ -45,7 +45,7 @@ def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
     plt.plot(df['Portfolio Value'], label='Portfolio Value', color='purple')
     plt.xlabel('Date')
     plt.ylabel('Portfolio Value ($)')
-    plt.title(f'Portfolio Value Over Time with 50/150 SMA Crossover Strategy ({ticker})')
+    plt.title(f'Portfolio Value Over Time ({ticker})')
     plt.legend()
     plt.grid()
     plt.tight_layout()
@@ -64,15 +64,15 @@ def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
     Trading Period: {start_date} to {end_date}
     Initial Investment: ${initial_budget}
     Final Portfolio Value: ${final_value:.2f}
-    Total Profit/Loss: ${profit_loss:.2f}
+    Profit/Loss: ${profit_loss:.2f}
     Percentage Return: {percentage_return:.2f}%
     """
 
     return plot_file, results
 
-# Define Gradio App
+
 with gr.Blocks() as app:
-    gr.Markdown("# SMA Crossover Trading Strategy Simulator")
+    gr.Markdown("## SMA Crossover Trading Strategy Simulator")
 
     with gr.Row():
         initial_budget = gr.Number(label="Initial Investment ($)", value=100)
@@ -93,7 +93,5 @@ with gr.Blocks() as app:
         outputs=[portfolio_graph, summary_text],
     )
 
-if __name__ == "__main__":
-    app.launch()
-
+app.launch()