Update main.py

main.py

@@ -36,17 +36,16 @@ def get_data_from_yahoo(ticker: str):
     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):    
+    api_key = 'nixt-7nixiWJtV1TwD3vp9K8WqLrgkIZmsXZ0gxBqrOSI1E3XPpkVSakyPYgMtWdtNKeBgStnPncgzpGqQzoG'
+    nixtla_client = NixtlaClient(api_key = api_key)
+## forecasting 3 months     
+    timegpt_fcst_df = nixtla_client.forecast(df=df, h=3, freq='MS', time_col='Date', target_col='Adj Close')
+        
+    return timegpt_fcst_df
        
     
 
@@ -55,8 +54,8 @@ def apikey(api_key):
 @app.get("/ticker/{ticker}")
 def read_item(ticker: str):
     data = get_data_from_yahoo(ticker)
-    key= apikey(api_key)
-    result = key.to_json()
+    pred = forecasting(data)
+    result = pred.to_json()
     return result