test3.py

from fastapi import FastAPI, File, UploadFile, HTTPException
from sqlalchemy import create_engine, Column, Integer, LargeBinary, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from pydantic import BaseModel

# SQLAlchemy setup
DATABASE_URL = "sqlite:///./test.db"
engine = create_engine(DATABASE_URL)
Base = declarative_base()

class Audio(Base):
    __tablename__ = "audios"

    id = Column(Integer, primary_key=True, index=True)
    name = Column(String, index=True)
    data = Column(LargeBinary)

Base.metadata.create_all(bind=engine)

app = FastAPI()

class AudioUpload(BaseModel):
    name: str
    audio: UploadFile

@app.post("/upload/")
async def upload_audio(audio_data: AudioUpload):
    db = SessionLocal()
    try:
        audio_content = audio_data.audio.file.read()


        db_audio = Audio(name=audio_data.name, data=audio_content)
        db.add(db_audio)
        db.commit()

        return {"message": "Audio uploaded successfully"}

    except Exception as e:
        raise HTTPException(status_code=500, detail="Internal Server Error")

    finally:
        db.close()

def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()