database.py

from sqlalchemy import create_engine, Column, Integer, String, DateTime, Float
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from datetime import datetime

# Database URL, change as needed (e.g., to PostgreSQL or MySQL)
DATABASE_URL = "sqlite:///database.db"

# Create the database engine
engine = create_engine(DATABASE_URL, echo=True)

# Create a base class for models
Base = declarative_base()

# Define your models
class Prompt(Base):
    __tablename__ = 'prompts'
    id = Column(Integer, primary_key=True, index=True)
    user_id = Column(Integer)
    prompt = Column(String(1000))
    link = Column(String, nullable=True)
    language = Column(String , nullable=True)
    thumbnail= Column(String, nullable=True)
    cost = Column(Float, nullable=True)
    created_at = Column(DateTime, default=datetime.utcnow)

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True, index=True)
    user_id = Column(Integer)
    prompt = Column(String(1000))
    link = Column(String, nullable=True)
    cost = Column(Float, nullable=True)
    created_at = Column(DateTime, default=datetime.utcnow)


def create_db_and_tables():
    Base.metadata.create_all(engine)


SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

def get_session():
    with SessionLocal() as session:
        yield session