like dislike saving

app.py

@@ -1,5 +1,6 @@
 import gradio
 import yaml
+from pathlib import Path
 
 from mtg.bot import MagicGPT
 from mtg.history.chat_history import ChatHistory
@@ -38,7 +39,7 @@ def get_magic_bot() -> MagicGPT:
         model=MODEL_NAME,
         temperature_deck_building=0.7,
         max_token_limit=1000,
-        data_filepath=config.get("message_filepath", "data/messages"),
+        data_filepath=Path(config.get("message_filepath", "data/messages")),
     )
     return magic_bot
 

mtg/bot/magic_gpt.py

@@ -32,8 +32,8 @@ def __init__(
         )
 
         self.data_filepath: Path = data_filepath
-        (self.data_filepath / "liked").mkdir(exist_ok=True)
-        (self.data_filepath / "disliked").mkdir(exist_ok=True)
+        (self.data_filepath / "liked").mkdir(exist_ok=True, parents=True)
+        (self.data_filepath / "disliked").mkdir(exist_ok=True, parents=True)
 
         self.chat_history: ChatHistory = chat_history
         self.memory = memory

mtg/history/chat_history.py

@@ -186,7 +186,7 @@ def create_message(self, text: str, message_type: MessageType) -> Message:
         )
 
         # add additional cards
-        if message_type == "deckbuilding":
+        if message_type == MessageType.DECKBUILDING:
             message = self.add_additional_cards(message=message, max_number_of_cards=10)
 
         logger.info(
@@ -205,6 +205,7 @@ def add_additional_cards(
         threshold: float = 0.5,
         lasso_threshold: float = 0.03,
     ) -> Message:
+
         additional_cards = []
         for card in message.cards:
             # for each card in message get max_number_of_cards
@@ -218,13 +219,25 @@ def add_additional_cards(
                 )
             )
 
+        # from message
+        additional_cards.extend(
+            self.data_service.get_cards(
+                message.text,
+                k=max(10, max_number_of_cards),
+                threshold=threshold,
+                lasso_threshold=lasso_threshold,
+                sample_results=True,
+            )
+        )
+
         # choose cards
         cards = []
         for card in additional_cards:
             if card not in message.cards and not card in cards:
                 cards.append(card)
         cards = random.choices(cards, k=min(len(cards), max_number_of_cards))
 
+        # add additional cards
         message.cards.extend(cards)
         logger.debug(f"added {len(cards)} additional cards")
 

mtg/objects/card.py

@@ -22,6 +22,9 @@ class Card(BaseModel):
     def __repr__(self) -> str:
         return f"Card({self.name})"
 
+    def to_dict(self) -> dict:
+        return self.model_dump()
+
     def to_text(self, include_price: bool = True):
         """parse card data to text format"""
         text = []

mtg/objects/document.py

@@ -10,3 +10,6 @@ class Document(BaseModel):
 
     def __repr__(self):
         return f"Document({self.name})"
+
+    def to_dict(self):
+        return self.model_dump()

mtg/objects/message.py

@@ -24,6 +24,6 @@ def to_dict(self):
         return {
             "text": self.text,
             "type": str(self.type),
-            "cards": [asdict(card) for card in self.cards],
-            "rules": [asdict(rule) for rule in self.rules],
+            "cards": [card.to_dict() for card in self.cards],
+            "rules": [rule.to_dict() for rule in self.rules],
         }