File size: 1,562 Bytes
0483de7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | from sqlalchemy import create_engine, Column, Integer, String, DateTime, Text
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from datetime import datetime
import os
from dotenv import load_dotenv
load_dotenv()
DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./complaints.db")
engine = create_engine(DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base = declarative_base()
class Complaint(Base):
__tablename__ = "complaints"
id = Column(Integer, primary_key=True, index=True)
complaint_text = Column(Text, nullable=False)
image_path = Column(String, nullable=False)
ai_response = Column(Text, nullable=False)
created_at = Column(DateTime, default=datetime.utcnow)
status = Column(String, default="resolved")
def create_tables():
Base.metadata.create_all(bind=engine)
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()
def save_complaint(complaint_text: str, image_path: str, ai_response: str):
db = SessionLocal()
try:
complaint = Complaint(
complaint_text=complaint_text,
image_path=image_path,
ai_response=ai_response,
status="resolved"
)
db.add(complaint)
db.commit()
db.refresh(complaint)
return complaint
finally:
db.close()
def get_all_complaints():
db = SessionLocal()
try:
return db.query(Complaint).all()
finally:
db.close() |