Update app.py

app.py

@@ -3,9 +3,17 @@ import yfinance as yf
 import numpy as np
 import matplotlib.pyplot as plt
 import gradio as gr
+import io
 
 def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
-    df = yf.download(ticker, start=start_date, end=end_date, progress=False)
+    print("Fetching data...")
+    try:
+        df = yf.download(ticker, start=start_date, end=end_date, progress=False)
+        if df.empty:
+            return None, "No data available for the specified ticker and date range.", None
+    except Exception as e:
+        return None, f"Error fetching data: {str(e)}", None
+
     df = df[['Close']]
 
     df['SMA_50'] = df['Close'].rolling(window=50).mean()
@@ -20,7 +28,10 @@ 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
         if row['Position'] == 1 and cash > 0:
             shares = cash / row['Close']
             cash = 0
@@ -43,8 +54,9 @@ def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
     plt.grid()
     plt.tight_layout()
 
-    plot_file = "portfolio_value_plot.png"
-    plt.savefig(plot_file)
+    plot_file = io.BytesIO()
+    plt.savefig(plot_file, format='png')
+    plot_file.seek(0)
     plt.close()
 
     final_value = portfolio_values[-1]
@@ -60,11 +72,7 @@ def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
     Percentage Return: {percentage_return:.2f}%
     """
 
-    results_file = "simulation_results.txt"
-    with open(results_file, "w") as f:
-        f.write(results)
-
-    return plot_file, results, results_file
+    return plot_file, results, None
 
 with gr.Blocks() as app:
     gr.Markdown("# SMA Crossover Trading Strategy Simulator")
@@ -84,7 +92,6 @@ with gr.Blocks() as app:
             with gr.Row():
                 portfolio_graph = gr.Image(label="Portfolio Value Over Time")
                 summary_text = gr.Textbox(label="Simulation Summary", lines=8)
-                download_button = gr.File(label="Download Results (.txt)")
 
         with gr.Tab("Instructions"):
             gr.Markdown("""
@@ -93,13 +100,13 @@ with gr.Blocks() as app:
             2. Specify the trading period (start and end dates).
             3. Select a stock ticker symbol (e.g., SPY, TSLA, GOOGL).
             4. Click "Run Simulation" to visualize the portfolio value over time and view a summary of results.
-            5. Download the results as a `.txt` file using the download button.
             """)
 
     run_button.click(
         sma_crossover_strategy,
         inputs=[initial_budget, start_date, end_date, ticker],
-        outputs=[portfolio_graph, summary_text, download_button],
+        outputs=[portfolio_graph, summary_text],
     )
 
 app.launch()
+