fix: continue session on another device, Fixes #21

Author: Tobi-De

README.md

@@ -3,5 +3,3 @@
 [![fuzzy-couscous](https://img.shields.io/badge/built%20with-fuzzy--couscous-success)](https://github.com/Tobi-De/fuzzy-couscous)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![linting: pylint](https://img.shields.io/badge/linting-pylint-yellowgreen)](https://github.com/PyCQA/pylint)
-
-

config/settings.py

@@ -205,6 +205,13 @@
 )
 SERVER_EMAIL = env("DJANGO_SERVER_EMAIL", default=DEFAULT_FROM_EMAIL)
 
+CACHES = {
+    "default": {
+        "BACKEND": "django.core.cache.backends.db.DatabaseCache",
+        "LOCATION": "leerming_cache_table",
+    }
+}
+
 if DJANGO_ENV == "production":
     import sentry_sdk
     from sentry_sdk.integrations.django import DjangoIntegration

docker/entrypoint.sh

@@ -3,6 +3,7 @@
 python manage.py compress
 python manage.py collectstatic --noinput
 python manage.py migrate
+python manage.py createcachetable
 python manage.py installwatson
 python manage.py makesuperuser
 #gunicorn config.wsgi --config="docker/gunicorn.conf.py"

leerming/flashcards/forms.py

@@ -120,8 +120,8 @@ def clean(self):
 
 
 class LLMFlashCard(forms.ModelForm):
+    card_type = forms.CharField(label="", widget=forms.HiddenInput())
 
-    card_type = forms.CharField(label="",widget=forms.HiddenInput())
     class Meta:
         model = FlashCard
         fields = ("question", "answer", "card_type")
@@ -144,5 +144,3 @@ def clean(self):
                 ),
             )
         return cleaned_data
-    
-

leerming/flashcards/llm_utils/__init__.py

@@ -2,18 +2,15 @@
 from dataclasses import dataclass
 from functools import cached_property
 
-
 from django.http import HttpRequest
 from langchain.llms import OpenAI
 
 from ..models import FlashCard
-from .prompts import (
-    clozed_delete_template,
-    front_back_template,
-    pipe_separated_list_output_parser,
-    double_pipe_separated_list_output_parser,
-    ParsedCard,
-)
+from .prompts import clozed_delete_template
+from .prompts import double_pipe_separated_list_output_parser
+from .prompts import front_back_template
+from .prompts import ParsedCard
+from .prompts import pipe_separated_list_output_parser
 
 
 LLM = OpenAI()
@@ -64,7 +61,9 @@ def make_flashcards_from(
             "min_result": max_result,
         }
     )
-    return [LLMFlashCard(**c, card_type=card_type, topic_id=topic_id) for c in parsed_cards]
+    return [
+        LLMFlashCard(**c, card_type=card_type, topic_id=topic_id) for c in parsed_cards
+    ]
 
 
 def save_llm_flashcards_to_session(

leerming/flashcards/llm_utils/prompts.py

@@ -1,8 +1,9 @@
 import re
 import uuid
+from typing import TypedDict
+
 from langchain.prompts import PromptTemplate
 from langchain.schema import BaseOutputParser
-from typing import TypedDict
 
 
 """
@@ -72,26 +73,27 @@
 
 
 class ParsedCard(TypedDict):
-    id:str
-    question:str
+    id: str
+    question: str
     answer: str
 
 
 class PipeSeparatedListOutputParser(BaseOutputParser):
     """Parse the output of an LLM call to a comma-separated list."""
 
-
-    def parse(self, text: str)->list[ParsedCard]:
+    def parse(self, text: str) -> list[ParsedCard]:
         """Parse the output of an LLM call."""
-        statements =  text.strip().split("|")
+        statements = text.strip().split("|")
         flashcards = []
         for statement in statements:
-            statement, clozed_deletions = self.extract_clozed_deletions(statement).values()
+            statement, clozed_deletions = self.extract_clozed_deletions(
+                statement
+            ).values()
             flashcards.extend(
                 {
-                    "id":str(uuid.uuid4()),
-                    "question":statement,
-                    "answer":cz,
+                    "id": str(uuid.uuid4()),
+                    "question": statement,
+                    "answer": cz,
                 }
                 for cz in clozed_deletions
             )
@@ -103,25 +105,31 @@ def extract_clozed_deletions(statement):
         cloze_matches = re.findall(cloze_pattern, statement)
         cleaned_statement = re.sub(r"{c\d+::(.*?)}", r"\1", statement)
 
-        return {"statement": cleaned_statement.strip(), "clozed_deletions": cloze_matches}
-    
+        return {
+            "statement": cleaned_statement.strip(),
+            "clozed_deletions": cloze_matches,
+        }
+
+
 pipe_separated_list_output_parser = PipeSeparatedListOutputParser()
-    
-class DoublePipeSeparatedListOutputParser(BaseOutputParser):
 
-    def parse(self, text: str)->list[ParsedCard]:
+
+class DoublePipeSeparatedListOutputParser(BaseOutputParser):
+    def parse(self, text: str) -> list[ParsedCard]:
         result = text.strip().split("||")
-        print(result)
         flashcards = []
         for qa in result:
             try:
-                flashcards.append({
-                    "id":str(uuid.uuid4()),
-                    "question":qa.split("|")[0],
-                    "answer":qa.split("|")[1],
-                })
+                flashcards.append(
+                    {
+                        "id": str(uuid.uuid4()),
+                        "question": qa.split("|")[0],
+                        "answer": qa.split("|")[1],
+                    }
+                )
             except IndexError:
                 continue
         return flashcards
-    
-double_pipe_separated_list_output_parser = DoublePipeSeparatedListOutputParser()
+
+
+double_pipe_separated_list_output_parser = DoublePipeSeparatedListOutputParser()

leerming/flashcards/views.py

@@ -132,7 +132,11 @@ def edit_llm_flashcard(request: HttpRequest, id: str):
     flashcard = [flashcard for flashcard in flashcards if flashcard.id == id][0]
     form = LLMFlashCard(
         request.POST or None,
-        initial={"question": flashcard.question, "answer": flashcard.answer, "card_type": flashcard.card_type},
+        initial={
+            "question": flashcard.question,
+            "answer": flashcard.answer,
+            "card_type": flashcard.card_type,
+        },
     )
     if request.POST and form.is_valid():
         flashcard.question = form.cleaned_data["question"]