added app.py

app.py

@@ -1,84 +1,7 @@
-from fastapi import FastAPI, Body
-from pydantic import BaseModel
-from typing import List, Optional
-
-import time
-
+from fastapi import FastAPI
 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)
-