Initial commit: Whisper API with FastAPI, GPU support and Admin Dashboard

src/database/__init__.py

@@ -0,0 +1,32 @@
+from sqlalchemy import create_engine
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+
+from src.config import settings
+
+# Create engine
+engine = create_engine(
+    settings.database_url,
+    connect_args={"check_same_thread": False} if "sqlite" in settings.database_url else {}
+)
+
+# Create session factory
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+# Base class for models
+Base = declarative_base()
+
+
+async def init_db():
+    """Initialize database tables"""
+    from src.database import models
+    Base.metadata.create_all(bind=engine)
+
+
+def get_db():
+    """Dependency to get database session"""
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()

src/database/db.py

@@ -0,0 +1,32 @@
+from sqlalchemy import create_engine
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+
+from src.config import settings
+
+# Create engine
+engine = create_engine(
+    settings.database_url,
+    connect_args={"check_same_thread": False} if "sqlite" in settings.database_url else {}
+)
+
+# Create session factory
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+# Base class for models
+Base = declarative_base()
+
+
+async def init_db():
+    """Initialize database tables"""
+    from src.database import models
+    Base.metadata.create_all(bind=engine)
+
+
+def get_db():
+    """Dependency to get database session"""
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()

src/database/models.py

@@ -0,0 +1,30 @@
+from sqlalchemy import Column, Integer, String, DateTime, Boolean, Float, Text
+from sqlalchemy.sql import func
+from src.database.db import Base
+
+
+class ApiKey(Base):
+    __tablename__ = "api_keys"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    key_hash = Column(String(64), unique=True, index=True)
+    description = Column(String(255), nullable=True)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+    last_used_at = Column(DateTime(timezone=True), nullable=True)
+    is_active = Column(Boolean, default=True)
+    usage_count = Column(Integer, default=0)
+
+
+class UsageLog(Base):
+    __tablename__ = "usage_logs"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    api_key_id = Column(Integer, nullable=True)
+    endpoint = Column(String(100))
+    file_size_bytes = Column(Integer, nullable=True)
+    duration_seconds = Column(Float, nullable=True)
+    processing_time_ms = Column(Integer)
+    model_used = Column(String(50))
+    status = Column(String(20))  # success, error
+    error_message = Column(Text, nullable=True)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())