add project type

1 files changed, 3 insertions, 1 deletions

diff --git a/animism-align/cli/app/sql/models/annotation.py b/animism-align/cli/app/sql/models/annotation.py
index 7363d16..750267a 100644
--- a/animism-align/cli/app/sql/models/annotation.py
+++ b/animism-align/cli/app/sql/models/annotation.py
@@ -13,6 +13,7 @@ class Annotation(Base):
   """Table for storing references to annotations"""
   __tablename__ = 'annotation'
   id = Column(Integer, primary_key=True)
+  episode_id = Column(Integer)
   type = Column(String(16, convert_unicode=True), nullable=False)
   paragraph_id = Column(Integer, nullable=True)
   start_ts = Column(Float, nullable=False)
@@ -23,8 +24,9 @@ class Annotation(Base):
   def toJSON(self):
     return {
       'id': self.id,
-      'type': self.type,
+      'episode_id': self.episode_id,
       'paragraph_id': self.paragraph_id,
+      'type': self.type,
       'start_ts': self.start_ts,
       'end_ts': self.end_ts,
       'text': self.text,