main.py

from fastapi import FastAPI, Body
from starlette.middleware.cors import CORSMiddleware
from fastapi.openapi.utils import get_openapi
from typing import List
import os
import collections
from api.models import Candidate, Article, Program
from api.models import init_programs, init_candidate
from api.database import query_similar_background_api, query_target_school_api
from api.parser import parse_request
from config import settings

app = FastAPI()
if settings.BACKEND_CORS_ORIGINS:
    app.add_middleware(
        CORSMiddleware,
        allow_origins=[str(origin) for origin in settings.BACKEND_CORS_ORIGINS],
    )


@app.on_event("startup")
def start_up_fn():
    # Run something on startup
    pass


@app.get("/")
def read_root():
    return {"docs": "Try out APIs and read documentation at '/docs'"}


@app.post("/admission", response_model=List[Article], tags=['admission'])
def list_programs(student: Candidate) -> List[Article]:
    result = []
    try:
        query_dict = parse_request(student, article_type="ADMISSION")
        articles = query_similar_background_api(query_dict)
        print('Query:', query_dict)
        max_score = articles[0].score if articles and articles[0].score else 0
        for idx, article in enumerate(articles[:settings.MAX_NUMBER_OF_ARTICLES]):
            if article.score is None or (len(articles) > 100 and article.score < max_score // 2):
                break
            programs = init_programs(article)
            result.append(init_candidate(article, programs))
        print('Results:', len(articles), len(result))
    except Exception as error:
        print(error)
    return result


@app.post("/school", response_model=List[Article], tags=['school'])
def list_target_school_info(student: Candidate) -> List[Article]:
    result = []
    try:
        query_dict = parse_request(student, article_type="ADMISSION")
        articles = query_target_school_api(query_dict)
        max_score = articles[0].score if articles and articles[0].score else 0
        for idx, article in enumerate(articles[:settings.MAX_NUMBER_OF_ARTICLES]):
            if article.score is None or (len(articles) > 100 and article.score != max_score):
                break
            programs = init_programs(article)
            result.append(init_candidate(article, programs))
        print(query_dict, len(articles), len(result))
    except Exception as error:
        print(error)
    return result


def custom_openapi():
    if app.openapi_schema:
        return app.openapi_schema

    openapi_schema = get_openapi(
        title="PTT StudyAbroad API",
        version="0.1.0",
        description="List of APIs to search CS related articles on PTT StudyAbroad board",
        routes=app.routes,
    )
    openapi_schema['tags'] = [
        {"name": "admission", "description": "Get a list of admission articles with similar background information"},
        {"name": "school", "description": "Get a list of admission articles for a target school program"},
    ]
    app.openapi_schema = openapi_schema
    return app.openapi_schema


app.openapi = custom_openapi