add write audio to db

autoLoader/database/__init__.py

@@ -1,7 +1,8 @@
 from sqlalchemy import create_engine
 from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import sessionmaker
+from sqlalchemy.orm import sessionmaker, Session
 from autoLoader.config import DATABASE_URL
+from contextlib import contextmanager
 
 # Создание engine
 engine = create_engine(
@@ -12,11 +13,11 @@ engine = create_engine(
 # SessionLocal
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
-# Base
-Base = declarative_base()
+# Base (используем тот же Base, что и в apiApp.database)
+from apiApp.database import Base
 
 
-# Зависимость для получения сессии БД
+# Зависимость для получения сессии БД (для FastAPI)
 def get_db():
     db = SessionLocal()
     try:
@@ -25,7 +26,22 @@ def get_db():
         db.close()
 
 
-# from apiApp.database.Operator import Operator
+# Контекстный менеджер для использования в loader
+@contextmanager
+def get_db_session():
+    """Контекстный менеджер для работы с БД в loader"""
+    db = SessionLocal()
+    try:
+        yield db
+        db.commit()
+    except Exception as e:
+        db.rollback()
+        raise e
+    finally:
+        db.close()
+
+
+# Импортируем модели из apiApp.database
 from apiApp.database.Audio import Audio
 from apiApp.database.AiConclusion import AiConclusion
-from apiApp.database.ConclusionVersion import ConclusionVersion
+from apiApp.database.ConclusionVersion import ConclusionVersion