Check of poems in database added in function /api/user/poem

back/api.py

@@ -34,8 +34,7 @@ if not os.path.exists("./uploads"):
 app.mount("/uploads", StaticFiles(directory = "./uploads"))
 
 # Записываем стихи в базу данных, если их еще нет (запускать только если стихов в базе нет).
-# from .db import database
-# service.add_poems_to_db(database)
+from .db import database
 
 
 @app.get("/api/announcements", response_model=List[schemas.Announcement])#адрес объявлений 
@@ -193,7 +192,7 @@ def add_points(user_id: int, db: Annotated[Session, Depends(utils.get_db)]):
 def poems_to_front(db: Annotated[Session, Depends(utils.get_db)]): # db: Annotated[Session, Depends(utils.get_db)] 
     num_of_poems = db.query(models.Poems).count()  # определяем кол-во стихов в бд
     if num_of_poems < 1:
-        raise HTTPException(status_code=404, detail="Poems not found")
+        service.add_poems_to_db(database) # добавляем поэмы в базу данных
     rand_id = random.randint(1, num_of_poems)  # генерируем номер стихотворения
     poem = db.query(models.Poems).filter(models.Poems.id == rand_id).first() # находим стих в бд
     if not poem: