models.py

from datetime import datetime
from app import db
from flask_login import UserMixin

class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(64), unique=True, nullable=False)
    email = db.Column(db.String(120), unique=True, nullable=False)
    edsl_api_key = db.Column(db.String(256))
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    
    # Relationship to review sessions
    review_sessions = db.relationship('ReviewSession', backref='user', lazy=True, cascade='all, delete-orphan')

class ReviewSession(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(200), nullable=False)
    paper_filename = db.Column(db.String(200), nullable=False)
    paper_path = db.Column(db.String(500), nullable=False)
    status = db.Column(db.String(50), default='uploaded')  # uploaded, processing, completed, error
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    completed_at = db.Column(db.DateTime)
    
    # Foreign key to user
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
    
    # Review content
    referee_reports = db.Column(db.Text)
    author_response = db.Column(db.Text)
    referee_rebuttals = db.Column(db.Text)
    
    # Processing metadata
    error_message = db.Column(db.Text)
    processing_stage = db.Column(db.String(100))  # referee_reports, author_response, referee_rebuttals