Update app.py

app.py

@@ -3,7 +3,7 @@ import re
 import torch
 from threading import Thread
 from typing import Iterator
-from mongoengine import connect, Document, StringField, SequenceField
+from mongoengine import connect, Document, StringField, SequenceField, DateTimeField
 import gradio as gr
 import spaces
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, TextIteratorStreamer
@@ -51,6 +51,7 @@ class Story(Document):
     message = StringField()
     content = StringField()
     story_id = SequenceField(primary_key=True)
+    created_at = DateTimeField(default=datetime.datetime.now)
 
 # Utility function for prompts
 def make_prompt(entry):