diff --git a/main.py b/main.py index 6d12b31..1f134a5 100644 --- a/main.py +++ b/main.py @@ -3,7 +3,8 @@ import logging from contextlib import asynccontextmanager from database import create_tables, delete_tables -from router import router as questions_router +from router_questions import router as questions_router +from router_class import router as class_router # Настройка логирования logging.basicConfig(level=logging.INFO) @@ -20,3 +21,4 @@ async def lifespan(app: FastAPI): app = FastAPI(lifespan=lifespan) app.include_router(questions_router) +app.include_router(class_router) diff --git a/router_class.py b/router_class.py new file mode 100644 index 0000000..9e9c1c5 --- /dev/null +++ b/router_class.py @@ -0,0 +1,25 @@ +from typing import Annotated +from fastapi import APIRouter, Depends +from typing import List + +from enums import TypeMood, TypeModel +from repository import QuestionRepository +from schemas import SQuestionAdd, SQuestion, SQuestionId + +router = APIRouter( + prefix="/class", + tags=["Class"], +) + +@router.get("/negative") +async def get_class_names() -> List[str]: + with open(".//neural_network/classification/class_names_negative.txt", "r", encoding="utf-8") as file: + class_names = [line.strip() for line in file.readlines()] + return class_names + +@router.get("/positive") +async def get_class_names() -> List[str]: + with open(".//neural_network/classification/class_names_positive.txt", "r", encoding="utf-8") as file: + class_names = [line.strip() for line in file.readlines()] + return class_names + diff --git a/router.py b/router_questions.py similarity index 66% rename from router.py rename to router_questions.py index 632878d..705f9f3 100644 --- a/router.py +++ b/router_questions.py @@ -11,18 +11,6 @@ router = APIRouter( tags=["Questions"], ) -@router.get("/class_negative") -async def get_class_names() -> List[str]: - with open(".//neural_network/classification/class_names_negative.txt", "r", encoding="utf-8") as file: - class_names = [line.strip() for line in file.readlines()] - return class_names - -@router.get("/class_positive") -async def get_class_names() -> List[str]: - with open(".//neural_network/classification/class_names_positive.txt", "r", encoding="utf-8") as file: - class_names = [line.strip() for line in file.readlines()] - return class_names - @router.post("") async def add_question( question: Annotated[SQuestionAdd, Depends()], @@ -42,3 +30,4 @@ async def get_questions() -> list[SQuestion]: async def get_questions_by_email(email_user: str) -> list[SQuestion]: questions = await QuestionRepository.find_by_email(email_user) return questions +