Update app.py

app.py

@@ -6,6 +6,17 @@ import os
 import time
 import torch
 from random import choice
+# Replace Conversation with custom class
+class ArkanaMemory:
+    def __init__(self):
+        self.history = []
+        
+    def add_exchange(self, user, arkana):
+        self.history.append({
+            "user": user,
+            "arkana": arkana,
+            "timestamp": datetime.now().isoformat()
+        })
 
 # Configuration
 MODEL_NAME = "google/flan-t5-large"
@@ -42,7 +53,7 @@ generator = pipeline(
     device=DEVICE,
     torch_dtype=torch.float16
 )
-conversation_memory = Conversation()
+conversation_memory = ArkanaMemory()
 
 # Voice Functions
 def text_to_speech(text):