Update main.py

main.py

@@ -130,3 +130,44 @@ def read_item(ticker: str):
     stock_data = get_stock_data(ticker)
     result = stock_data.to_json(orient="records")
     return result
+
+## TimeGPT 
+    
+@app.get("/ticker/{ticker}")
+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(key_valid,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')
+