Update app.py

app.py

@@ -52,7 +52,7 @@ def stock_forecast(ticker, start_date, end_date):
             inputs=df,
             freq="D",  # Daily frequency
             value_name="y",
-            num_jobs=-1,
+            num_jobs=2,
         )
 
         # Combine actual and forecasted data for plotting
@@ -62,8 +62,12 @@ def stock_forecast(ticker, start_date, end_date):
         fig = px.line(combined_df, x='ds', y=['y', 'timesfm'], labels={'value': 'Price', 'ds': 'Date'}, 
                       title=f'{ticker} Stock Price Forecast')
         
-        # Enhance interactivity
+        # Customize the plot for a dark theme
         fig.update_layout(
+            template="plotly_dark",  # Use a dark theme
+            plot_bgcolor='rgba(0, 0, 0, 0)',  # Transparent plot background
+            paper_bgcolor='rgba(0, 0, 0, 0)',  # Transparent paper background
+            font=dict(color='white'),  # White font color
             legend_title_text='Type',
             hovermode='x unified',  # Show hover info for all series at the same x-value
             xaxis=dict(
@@ -74,12 +78,34 @@ def stock_forecast(ticker, start_date, end_date):
                         dict(count=1, label="YTD", step="year", stepmode="todate"),
                         dict(count=1, label="1y", step="year", stepmode="backward"),
                         dict(step="all")
-                    ])
+                    ]),
+                    bgcolor='black',  # Dark background for range selector
+                    font=dict(color='white')  # White font for range selector
                 ),
                 rangeslider=dict(visible=True),  # Add a range slider
-                type="date"
+                type="date",
+                gridcolor='gray',  # Grid lines color
+                linecolor='gray',  # Axis line color
+                tickfont=dict(color='white')  # White tick labels
             ),
-            yaxis=dict(title="Price (USD)")
+            yaxis=dict(
+                title="Price (USD)",
+                gridcolor='gray',  # Grid lines color
+                linecolor='gray',  # Axis line color
+                tickfont=dict(color='white')  # White tick labels
+            )
+        )
+
+        # Customize line colors
+        fig.update_traces(
+            line=dict(width=2),  # Thicker lines for better visibility
+            selector=dict(name='y'),  # Actual prices
+            line_color='blue'  # Blue for actual prices
+        )
+        fig.update_traces(
+            line=dict(width=2),  # Thicker lines for better visibility
+            selector=dict(name='timesfm'),  # Forecasted prices
+            line_color='purple'  # Purple for forecasted prices
         )
 
         # Add custom hover data
@@ -93,10 +119,14 @@ def stock_forecast(ticker, start_date, end_date):
         # Return an empty Plotly figure with an error message
         error_fig = go.Figure()
         error_fig.update_layout(
+            template="plotly_dark",  # Use a dark theme
             title=f"Error: {str(e)}",
             xaxis_title="Date",
             yaxis_title="Price",
-            annotations=[dict(text="No data available", x=0.5, y=0.5, showarrow=False)]
+            annotations=[dict(text="No data available", x=0.5, y=0.5, showarrow=False, font=dict(color='white'))],
+            plot_bgcolor='rgba(0, 0, 0, 0)',  # Transparent plot background
+            paper_bgcolor='rgba(0, 0, 0, 0)',  # Transparent paper background
+            font=dict(color='white')  # White font color
         )
         return error_fig
 
@@ -108,7 +138,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         ticker_input = gr.Textbox(label="Enter Stock Ticker", value="AAPL")
         start_date_input = gr.Textbox(label="Enter Start Date (YYYY-MM-DD)", value="2022-01-01")
-        end_date_input = gr.Textbox(label="Enter End Date (YYYY-MM-DD)", value="2025-01-01")
+        end_date_input = gr.Textbox(label="Enter End Date (YYYY-MM-DD)", value="2026-01-01")
     
     submit_button = gr.Button("Enter")
     plot_output = gr.Plot()