feat: split message class

app.py

@@ -1,21 +1,12 @@
 import os
-from dataclasses import dataclass
-from typing import Literal
-import streamlit as st
 
+import streamlit as st
+import streamlit.components.v1 as components
 from langchain import OpenAI
 from langchain.callbacks import get_openai_callback
 from langchain.chains import ConversationChain
 from langchain.chains.conversation.memory import ConversationSummaryMemory
-import streamlit.components.v1 as components
-
-
-@dataclass
-class Message:
-    """Class for keeping track of a chat message."""
-
-    origin: Literal["human", "ai"]
-    message: str
+from message import Message
 
 
 def load_css():

message.py

@@ -0,0 +1,10 @@
+from dataclasses import dataclass
+from typing import Literal
+
+
+@dataclass
+class Message:
+    """Class for keeping track of a chat message."""
+
+    origin: Literal["human", "ai"]
+    message: str