From 1e3a2f3efa6ffdcc26af6a3eb158fe8e165ebf4c Mon Sep 17 00:00:00 2001
From: maksim <kashin20031984@mail.ru>
Date: Sun, 2 Jun 2024 17:33:50 +0400
Subject: [PATCH] =?UTF-8?q?=D0=A0=D0=B0=D0=B7=D0=B4=D1=80=D0=BE=D0=B1?=
 =?UTF-8?q?=D0=B8=D0=BB=20=D1=80=D0=BE=D1=83=D1=82=D0=B5=D1=80=20=D0=B0?=
 =?UTF-8?q?=D0=B4=D0=B5=D0=BA=D0=B2=D0=B0=D1=82=D0=BD=D0=BE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py                          |  4 +++-
 router_class.py                  | 25 +++++++++++++++++++++++++
 router.py => router_questions.py | 13 +------------
 3 files changed, 29 insertions(+), 13 deletions(-)
 create mode 100644 router_class.py
 rename router.py => router_questions.py (66%)

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
+