CoolFace
Apppublic

Mery3391/YAMNET

sourceHugging Facemitupdated 4mo agoView on Hugging Face
0likes
database.py38 linesDownload Raw Back to root
1# database.py2from sqlalchemy import create_engine, Column, Integer, String, Float, DateTime3from sqlalchemy.ext.declarative import declarative_base4from sqlalchemy.orm import sessionmaker5from datetime import datetime6import os7 8# Créer la base de données SQLite9DATABASE_URL = "sqlite:///./voiceemotion.db"10engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})11Base = declarative_base()12 13class User(Base):14    __tablename__ = "users"15    id = Column(Integer, primary_key=True)16    email = Column(String, unique=True, index=True)17    password_hash = Column(String)18    name = Column(String)19    created_at = Column(DateTime, default=datetime.utcnow)20 21class AudioAnalysis(Base):22    __tablename__ = "analyses"23    id = Column(Integer, primary_key=True)24    user_id = Column(Integer, index=True)25    filename = Column(String)26    emotion = Column(String)27    confidence = Column(Float)  # ← Correction: Flout → Float28    angry_score = Column(Float, default=0)  # ← Correction: Flout → Float29    happy_score = Column(Float, default=0)30    neutral_score = Column(Float, default=0)31    sad_score = Column(Float, default=0)32    created_at = Column(DateTime, default=datetime.utcnow)33 34# Créer toutes les tables35Base.metadata.create_all(bind=engine)36 37# Session locale38SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)