File size: 2,022 Bytes
01e9350 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 | import gc
import logging
import os
import shutil
import tempfile
import time
from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware
from chartmetric_summary_generator3 import final_score_calculation
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
SESSIONS_ROOT = os.path.join(BASE_DIR, "sessions") # all per-request dirs live here
os.makedirs(SESSIONS_ROOT, exist_ok=True)
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - %(levelname)s - %(message)s",
)
logger = logging.getLogger(__name__)
def _cleanup_session(session_dir: str) -> None:
"""Remove the entire per-request temp directory, tolerating locked files."""
if not os.path.exists(session_dir):
return
try:
gc.collect()
time.sleep(0.1)
shutil.rmtree(session_dir, ignore_errors=True)
logger.info(f"Cleaned up session dir: {session_dir}")
except Exception as e:
logger.warning(f"Could not fully clean up {session_dir}: {e}")
@app.get("/")
def read_root():
return {"Hello": "200 ok"}
@app.post("/fanbase_score")
async def fanbase_score(request: Request):
session_dir = tempfile.mkdtemp(dir=SESSIONS_ROOT)
logger.info(f"Created session dir: {session_dir}")
try:
data = await request.json()
task_id = data["track_id"].strip()
logger.info(f"Processing start – task_id={task_id} session={session_dir}")
fanbase_score_count = final_score_calculation(task_id, session_dir)
logger.info(f"Fan Base Count: {fanbase_score_count}")
return {"fanbase_count": fanbase_score_count}
except Exception as e:
logger.error(f"Request failed: {e}")
return {"error": str(e)}
finally:
_cleanup_session(session_dir) |