added app.py

app.py

@@ -0,0 +1,84 @@
+from fastapi import FastAPI, Body
+from pydantic import BaseModel
+from typing import List, Optional
+
+import time
+
+import json
+import os
+import pandas as pd
+import getNewSec
+
+from tokenizeAPIHelpers import datasetHandler, tokenizeDir
+from hf_login import hf_login
+
+from huggingface_hub import HfApi, SpaceHardware
+
+app = FastAPI()
+
+class TickerItem(BaseModel):
+	ticker: str
+	cik: str
+	lastSync: Optional[str] = None
+
+@app.post("/downgradeTS")
+def downgradeTS():
+	token = os.getenv("HF_TOKEN")
+	if not token:
+		raise RuntimeError("HF_TOKEN not found in environment or .env file")
+
+	api = HfApi(token=token)
+	repo = "UMWRoom225/tokenizer"
+	api.request_space_hardware(repo_id=repo, hardware=SpaceHardware.CPU_BASIC)
+	return {"ok": True, "hardware": "CPU_BASIC"}
+
+@app.post("/upgradeTS")
+def upgradeTS():
+	token = os.getenv("HF_TOKEN")
+	if not token:
+		raise RuntimeError("HF_TOKEN not found in environment or .env file")
+
+	api = HfApi(token=token)
+	repo = "UMWRoom225/tokenizer"
+	api.request_space_hardware(repo_id=repo, hardware=SpaceHardware.CPU_UPGRADE)
+	return {"ok": True, "hardware": "CPU_UPGRADE"}
+
+@app.get("/")
+def statusCheck():
+	return {"status": "running"}
+
+@app.get("/health")
+def health():
+	token = os.getenv("HF_TOKEN")
+	if not token:
+		raise RuntimeError("HF_TOKEN not found in environment or .env file")
+
+	api = HfApi(token=token)
+	repo = "UMWRoom225/tokenizer"
+
+	state=api.get_space_runtime(repo_id=repo)
+	if(state.stage == "RUNNING"):
+		return {"ok": True}
+	else:
+		return {"ok":False}
+
+
+@app.post("/tokenize")
+def writeNewDocs(items: List[TickerItem] = Body(...)):
+	hf_login()
+	print("LETSSS GET TS",flush=True)
+	ticksDf = pd.DataFrame([i.model_dump() for i in items])
+	ticksDf = ticksDf.iloc[0:2]
+	print(ticksDf,flush=True)
+	toUpdate = getNewSec.main(ticksDf)
+	print(toUpdate,flush=True)
+	rtn = list(toUpdate)
+
+	print(rtn,flush=True)
+	time.sleep(10)
+	tokenizeDir(".", "tmp/", max_docs=300)
+	print("past tokenize",flush=True)
+	time.sleep(10)
+
+	return (rtn)
+