Update app.py

app.py

@@ -33,6 +33,10 @@ class HumanMessage:
     def __init__(self, content):
         self.content = content
 
+class AIMessage:
+    def __init__(self, content):
+        self.content = content
+
 
 def load_answer(question):