Update app.py

app.py

@@ -9,17 +9,17 @@ def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
     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
+            return None, "No data available for the specified ticker and date range."
     except Exception as e:
-        return None, f"Error fetching data: {str(e)}", None
+        return None, f"Error fetching data: {str(e)}"
 
     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
+    df.loc[df['SMA_50'] > df['SMA_150'], 'Signal'] = 1
+    df.loc[df['SMA_50'] < df['SMA_150'], 'Signal'] = -1
     df['Position'] = df['Signal'].diff()
 
     cash = initial_budget
@@ -38,7 +38,7 @@ def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
         portfolio_value = cash + (shares * row['Close'])
         portfolio_values.append(portfolio_value)
 
-    df = df.iloc[149:]
+    df = df.iloc[149:]  # Skip rows without SMA values
     df['Portfolio Value'] = portfolio_values[149:]
 
     plt.figure(figsize=(14, 8))
@@ -68,9 +68,9 @@ def sma_crossover_strategy(initial_budget, start_date, end_date, ticker):
     Percentage Return: {percentage_return:.2f}%
     """
 
-    return plot_file, results, None
+    return plot_file, results
 
-# Gradio Blocks Context
+# Define Gradio App
 with gr.Blocks() as app:
     gr.Markdown("# SMA Crossover Trading Strategy Simulator")
 
@@ -87,13 +87,11 @@ with gr.Blocks() as app:
     portfolio_graph = gr.Image(label="Portfolio Value Over Time")
     summary_text = gr.Textbox(label="Simulation Summary", lines=8)
 
-    # Ensure run_button.click() is defined here
     run_button.click(
         sma_crossover_strategy,
         inputs=[initial_budget, start_date, end_date, ticker],
         outputs=[portfolio_graph, summary_text],
     )
 
-app.launch(share=True)
-
+app.launch()  # No share=True