Update app.py

app.py

@@ -2,35 +2,42 @@ import gradio as gr
 import yfinance as yf
 from prophet import Prophet
 import pandas as pd
+from datetime import datetime
 import plotly.graph_objects as go
 
 def download_data(ticker, start_date='2010-01-01'):
+    """ 주식 데이터를 다운로드하고 포맷을 조정하는 함수 """
     data = yf.download(ticker, start=start_date)
     if data.empty:
         raise ValueError(f"No data returned for {ticker}")
     data.reset_index(inplace=True)
     if 'Adj Close' in data.columns:
-        data = data[['Date', 'Adj Close']].copy()
+        data = data[['Date', 'Adj Close']]
         data.rename(columns={'Date': 'ds', 'Adj Close': 'y'}, inplace=True)
     else:
         raise ValueError("Expected 'Adj Close' in columns")
-    data['ds'] = pd.to_datetime(data['ds'])
     return data
 
 def predict_future_prices(ticker, periods=1825):
     data = download_data(ticker)
-    model = Prophet(daily_seasonality=False, weekly_seasonality=True, yearly_seasonality=True)
+    
+    # Prophet 모델 생성 및 학습
+    model = Prophet(daily_seasonality=False, weekly_seasonality=False, yearly_seasonality=True)
     model.fit(data)
+    
+    # 미래 데이터 프레임 생성 및 예측
     future = model.make_future_dataframe(periods=periods, freq='D')
     forecast = model.predict(future)
     
+    # 예측 결과 그래프 생성
+    forecast['ds'] = forecast['ds'].dt.strftime('%Y-%m-%d')
     fig = go.Figure()
-    fig.add_trace(go.Scatter(x=data['ds'], y=data['y'], mode='lines', name='Actual (Black)', line=dict(color='black')))
     fig.add_trace(go.Scatter(x=forecast['ds'], y=forecast['yhat'], mode='lines', name='Forecast (Blue)'))
+    fig.add_trace(go.Scatter(x=data['ds'], y=data['y'], mode='lines', name='Actual (Black)', line=dict(color='black')))
     
-    # Make sure to return both the figure and the forecast data
     return fig, forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']]
 
+# Gradio 인터페이스 설정 및 실행
 with gr.Blocks() as app:
     with gr.Row():
         ticker_input = gr.Textbox(value="AAPL", label="Enter Stock Ticker for Forecast")