f

api/base_router.py

@@ -0,0 +1,11 @@
+from fastapi import APIRouter
+from api import prediction
+
+
+base_router = APIRouter()
+base_router.include_router(prediction.router)
+
+
+@base_router.get('/', name='Default page')
+async def default():
+    return 'Привет!'

api/prediction.py

@@ -0,0 +1,23 @@
+from transformers import pipeline
+import os
+from fastapi import APIRouter
+from models.prediction import Prediction
+
+from services.answers_service import AnswersService
+router = APIRouter(
+    prefix='/prediction',
+    tags=['prediction'],
+)
+
+
+nlp_task = os.getenv('NLP_TASK')
+model_name = os.getenv('MODEL')
+model = pipeline(nlp_task, model=model_name)
+
+
+@router.post("/get_prediction", name='Получение предсказания модели')
+def get_prediction(message_from_user: str):
+    output = model(message_from_user)
+    prediction = Prediction.from_output(dictionary=output[0])
+    answer = AnswersService.get_answer_by_id(prediction=prediction)
+    return answer

app.py

@@ -0,0 +1,19 @@
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+
+from api.base_router import base_router
+
+
+app = FastAPI(
+    docs_url='/docs',
+)
+
+app.include_router(base_router)
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)

core/settings.py

@@ -0,0 +1,2 @@
+class Config:
+    threshold = 0.5

main.py

@@ -0,0 +1,9 @@
+import uvicorn
+from dotenv import load_dotenv
+
+if __name__ == '__main__':
+    load_dotenv()
+    uvicorn.run(
+        'app:app',
+        reload=True,
+    )

models/prediction.py

@@ -0,0 +1,12 @@
+class Prediction:
+    def __init__(self, label: str, score: float):
+        self.label = label
+        self.score = score
+
+    @staticmethod
+    def from_output(dictionary: dict):
+        return Prediction(Prediction.get_label(dictionary["label"]), dictionary["score"])
+
+    @staticmethod
+    def get_label(label: str):
+        return int(label.split('_')[1])

repositories/answers_repository.py

@@ -0,0 +1,21 @@
+from repositories.repository import Repository
+
+
+class AnswersRepository(Repository):
+
+    answers = ["Общее количество бюджетных мест составляет 2 598 по 90 программам бакалавриата, специалитета и магистратуры. Из них на очной форме обучения: 1 724 – бакалавриат, 518 – магистратура, 296 – специалитет. 60 мест (42 – бакалавриат, 18 – магистратура) выделены для очно-заочной (вечерней) формы обучения. Еще 150 мест выделено в колледже технического университета – там доступно 9 специальностей и 1 профессия. Распределение мест представлено в информационных бюллетенях.",
+               "Среди испытаний: информационные технологии; безопасность жизнедеятельности; метрология, стандартизация и сертификация; история изобразительного искусства; основы педагогики и психологии; менеджмент; экономика организации.",
+               "Молодые люди, имеющие гражданство РФ и получившие среднее профессиональное образование, проходят вступительные испытания по предметам профессиональной направленности и по русскому языку либо по результатам ЕГЭ общеобразовательных предметов.",
+               "Абитуриенты могут получить дополнительные баллы за различные достижения, такие как аттестат или диплом с отличием, высокий балл по ЕГЭ или вступительным испытаниям, а также за знаки ГТО (золотой, серебряный, бронзовый). В этом году расширился перечень способов подтверждения достижений, и награды могут быть проверены онлайн. В общей сложности, абитуриенты могут получить до 10 дополнительных баллов, учитываемых при поступлении на бюджетное место. Полный список индивидуальных достижений доступен на сайте ОмГТУ в разделе «Абитуриенту».",
+               "Абитуриент политеха вправе подать документы на 5 направлений подготовки бакалавриата и программ специалитета. Он может участвовать в конкурсе и на бюджет, и на коммерческое место, если предусмотрены обе формы финансирования.",
+               "ОмГТУ имеет 6 общежитий, всего 2,374 спальных мест. Общежития предоставляются студентам и аспирантам с гражданством РФ, не имеющим омской прописки, а также иностранным студентам на время обучения. Иногородние студенты заочной формы обучения могут воспользоваться общежитием во время сессии и сдачи выпускных работ при наличии свободных мест. Также иногородние абитуриенты могут получить общежитие на время вступительных испытаний.",
+               "Необходимо подготовить следующие документы: документ, удостоверяющий личность; документ об образовании; СНИЛС (для граждан РФ); фото в электронном виде; документы об особых правах; индивидуальные достижения. Подать документы можно: лично в приемной комиссии (пр. Мира, 11, корпус 8, кабинет 115), онлайн через личный кабинет на сайте ОмГТУ во вкладке «Абитуриенту», с помощью суперсервиса «Поступление в вуз онлайн», с помощью почты/курьерской доставки на адрес приемной комиссии. ",
+               "Приемная кампания завершится 20 июля для тех, кто поступает по направлениям подготовки с дополнительными вступительными испытаниями и по результатам вступительных испытаний. Для абитуриентов, поступающих по результатам ЕГЭ, срок окончания - 25 июля. Абитуриенты магистратуры на бюджетные места могут подавать документы до 10 августа.",
+               "Нет, очередность значения не имеет. Распределение кандидатов на поступление будет осуществляться по указанным в заявлении приоритетным направлениям или специальностям.",
+               ]
+
+    default_answer = "Затрудняюсь ответить на ваш вопрос. Уточните свой вопрос, пожалуйста."
+
+    @staticmethod
+    def get_elem_by_id(id: int):
+        return AnswersRepository.answers[id]

repositories/intents_repository.py

@@ -0,0 +1,13 @@
+from repositories.repository import Repository
+
+
+class IntentsRepository(Repository):
+
+    intents = ['Бюджетные места', 'Вступительные испытания', 'Вуз после колледжа',
+               'Индивидуальные достижения', 'Количество специальностей',
+               'Общежитие', 'Подача документов', 'Приемная компания',
+               'Приоритет от подачи оригинала']
+
+    @staticmethod
+    def get_elem_by_id(id: int):
+        return IntentsRepository.intents[id]

repositories/repository.py

@@ -0,0 +1,4 @@
+class Repository:
+    @staticmethod
+    def get_elem_by_id(id: int):
+        pass

requirements.txt

@@ -0,0 +1,5 @@
+fastapi
+uvicorn
+transformers
+torch
+python-dotenv

services/answers_service.py

@@ -0,0 +1,14 @@
+from models.prediction import Prediction
+from core.settings import Config
+from repositories.answers_repository import AnswersRepository
+
+
+class AnswersService:
+
+    @staticmethod
+    def get_answer_by_id(prediction: Prediction):
+        if prediction.score > Config.threshold:
+
+            return AnswersRepository.get_elem_by_id(prediction.label)
+
+        return AnswersRepository.default_answer

services/intents_service.py

@@ -0,0 +1,8 @@
+from models.prediction import Prediction
+from repositories.intents_repository import IntentsRepository
+
+
+class IntentsService:
+    @staticmethod
+    def get_intent_by_id(prediction: Prediction):
+        return IntentsRepository.get_elem_by_id[prediction.label]