Update main.py

main.py

@@ -27,12 +27,53 @@ def read_root():
 
     }
 
+## TimeGPT 
+    
+
+def get_data_from_yahoo(ticker: str):
+    yf.pdr_override()
+    data = pdr.get_data_yahoo(ticker, start="2023-01-01", end=None)
+    df = pd.DataFrame(data).reset_index()
+    return df
+
+##TimeGPT APIkey validation
+
+api_key = 'nixt-7nixiWJtV1TwD3vp9K8WqLrgkIZmsXZ0gxBqrOSI1E3XPpkVSakyPYgMtWdtNKeBgStnPncgzpGqQzoG'
+
+def apikey(api_key):
+    nixtla_client = NixtlaClient(
+    api_key = api_key
+    )
+    key_valid = nixtla_client.validate_api_key()
+    return key_valid
+
+def forecasting(df):    
+   ## if key_valid == True:
+## forecasting 3 months     
+       timegpt_fcst_df = nixtla_client.forecast(df=df, h=3, freq='MS', time_col='Date', target_col='Adj Close')
+    # else:
+    #    timegpt_fcst_df= 'API key is not Valid'
+        
+    return timegpt_fcst_df, nixtla_client.plot(df, timegpt_fcst_df, time_col='Date', target_col='Adj Close')
+
+       
+    
+## Anomaly Detection
+
+def anomaly_detect(df):
+    timegpt_anomalies_df = nixtla_client.detect_anomalies(df, time_col='Date', target_col='Adj Close', freq='D', level = 95, date_features=True)
+    return nixtla_client.plot(df,
+             timegpt_anomalies_df,
+             time_col='Date',
+             target_col='Adj Close')
 
 
 @app.get("/ticker/{ticker}")
 def read_item(ticker: str):
-    ticker = ticker
-    return ticker
-
+    data = get_data_from_yahoo(ticker)
+    key = apikey(api_key)
+    stock_forecast = forecasting(df)
+    result = stock_forecast.to_json()
+    return result