Update app/models.py

app/models.py

@@ -4,8 +4,8 @@ from sqlalchemy.orm import relationship
 from app.database import Base
 import enum
 
-class Tone(str , enum.Enum):
-    neutral = "netural"
+class Tone(str, enum.Enum):
+    neutral = "neutral"
     formal = "formal"
     friendly = "friendly"
     investor = "investor"
@@ -16,7 +16,6 @@ class ContentType(str, enum.Enum):
     ir_letter = "ir_letter"
     investor_summary = "investor_summary"
 
-
 class DraftStatus(str, enum.Enum):
     draft = "draft"
     pending = "pending"
@@ -24,42 +23,36 @@ class DraftStatus(str, enum.Enum):
     scheduled = "scheduled"
     sent = "sent"
 
-
 class Draft(Base):
     __tablename__ = "drafts"
     id = Column(Integer, primary_key=True, index=True)
-    source_type = Column(String(20)) # pdf/url/text
-    source_ref = Column(Text) # URL or filename
+    source_type = Column(String(20))  # pdf/url/text
+    source_ref = Column(Text)         # URL or filename
     raw_text = Column(Text)
     content_type = Column(Enum(ContentType), nullable=False)
     tone = Column(Enum(Tone), nullable=False, default=Tone.neutral)
     title = Column(String(300))
-    body_md = Column(Text) # Markdown
+    body_md = Column(Text)            # Markdown
     status = Column(Enum(DraftStatus), nullable=False, default=DraftStatus.draft)
     created_at = Column(DateTime(timezone=True), server_default=func.now())
     updated_at = Column(DateTime(timezone=True), onupdate=func.now())
 
-
     # AB test subjects (A/B)
     subject_a = Column(String(300))
     subject_b = Column(String(300))
 
-
     # delivery options
-        deliver_email_list = Column(Text) # comma-separated emails
+    deliver_email_list = Column(Text)   # comma-separated emails
     deliver_x = Column(Boolean, default=False)
     deliver_note = Column(Boolean, default=False)
 
-
     deliveries = relationship("Delivery", back_populates="draft")
 
-
 class DeliveryChannel(str, enum.Enum):
     email = "email"
     x = "x"
     note = "note"
 
-
 class Delivery(Base):
     __tablename__ = "deliveries"
     id = Column(Integer, primary_key=True)
@@ -69,5 +62,4 @@ class Delivery(Base):
     result = Column(JSON)
     created_at = Column(DateTime(timezone=True), server_default=func.now())
 
-
-    draft = relationship("Draft", back_populates="deliveries")
+    draft = relationship("Draft", back_populates="deliveries")