Add app\models\writesage.py

app//models//writesage.py

@@ -0,0 +1,133 @@
+# app/models/writesage.py
+# Final Version: Fixed SQLAlchemy column type error
+# Timestamp: 2026-03-13
+
+from datetime import datetime
+from typing import Optional, List, Dict, Any
+import enum
+
+from sqlalchemy import (
+    String,
+    Integer,
+    DateTime,
+    ForeignKey,
+    Text,
+    JSON,
+    Enum,
+    Boolean,
+    Float,  # FIX: Imported Float for proper column type
+)
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+from sqlalchemy.sql import func
+
+from app.models.base import Base
+
+
+class ManuscriptStatus(str, enum.Enum):
+    """Lifecycle of a scholarly manuscript."""
+    DRAFT = "draft"
+    GENERATING = "generating"
+    REVIEW_REQUIRED = "review_required"
+    COMPLETED = "completed"
+
+
+class Manuscript(Base):
+    """
+    Core document being authored within the WriteSage workspace.
+    Links to the researcher's Saved Library and extraction context.
+    """
+    __tablename__ = "writesage_manuscripts"
+
+    id: Mapped[str] = mapped_column(String(64), primary_key=True)
+    user_id: Mapped[int] = mapped_column(
+        Integer, 
+        ForeignKey("users.id", ondelete="CASCADE"), 
+        nullable=False
+    )
+    title: Mapped[str] = mapped_column(String(255), nullable=False)
+    target_journal: Mapped[Optional[str]] = mapped_column(String(100))
+    status: Mapped[ManuscriptStatus] = mapped_column(
+        Enum(ManuscriptStatus), 
+        default=ManuscriptStatus.DRAFT
+    )
+
+    # Metadata context: PICO, effect sizes, and seeds used for grounding
+    context_papers: Mapped[str] = mapped_column(Text, default="[]")  # JSON list of DOIs
+    pico_context_id: Mapped[Optional[str]] = mapped_column(String(64))
+
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), 
+        server_default=func.now()
+    )
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), 
+        server_default=func.now(), 
+        onupdate=func.now()
+    )
+
+    # Relationships
+    user = relationship("User", back_populates="manuscripts")
+    sections = relationship(
+        "ManuscriptSection", 
+        back_populates="manuscript", 
+        cascade="all, delete-orphan"
+    )
+
+
+class ManuscriptSection(Base):
+    """
+    Individual components of the manuscript (IMRaD+ structure).
+    Supports dynamic architecture suggestion via StructGen.
+    """
+    __tablename__ = "writesage_sections"
+
+    id: Mapped[int] = mapped_column(
+        Integer, 
+        primary_key=True, 
+        autoincrement=True
+    )
+    manuscript_id: Mapped[str] = mapped_column(
+        ForeignKey("writesage_manuscripts.id", ondelete="CASCADE"),
+        nullable=False
+    )
+    name: Mapped[str] = mapped_column(String(100))  # e.g., "Methods", "Results"
+    order_index: Mapped[int] = mapped_column(Integer, nullable=False)
+    content: Mapped[Optional[str]] = mapped_column(Text)
+    is_ai_generated: Mapped[bool] = mapped_column(Boolean, default=True)
+
+    # Audit trail for Veritas Shield integrity tracking
+    # FIX: Changed from func.float() to Float for proper SQLAlchemy column type
+    originality_score: Mapped[Optional[float]] = mapped_column(Float, nullable=True)
+
+    # Relationships
+    manuscript = relationship("Manuscript", back_populates="sections")
+
+
+class JournalProfile(Base):
+    """
+    Cached journal intelligence for format adaptation.
+    Resolves structure, citation style, and word limits.
+    """
+    __tablename__ = "writesage_journal_profiles"
+
+    id: Mapped[int] = mapped_column(
+        Integer, 
+        primary_key=True, 
+        autoincrement=True
+    )
+    journal_name: Mapped[str] = mapped_column(
+        String(255), 
+        index=True, 
+        unique=True
+    )
+    issn: Mapped[Optional[str]] = mapped_column(String(20), index=True)
+
+    # Structural intelligence
+    required_sections: Mapped[str] = mapped_column(Text)  # JSON list
+    word_limit: Mapped[Optional[int]] = mapped_column(Integer)
+    citation_style: Mapped[str] = mapped_column(String(100))  # e.g., "Vancouver", "APA"
+    
+    last_updated: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), 
+        server_default=func.now()
+    )