Snapshot

frontend/src/components/App.tsx

@@ -25,7 +25,7 @@ export default function App() {
   const [words, setWords] = useState<Word[]>([])
   const [isLoading, setIsLoading] = useState(false)
 
-  const check = async () => {
+  const check = async (text: string) => {
     setIsLoading(true)
     try {
       const checkedWords = await checkText(text)
@@ -39,20 +39,24 @@ export default function App() {
   const toggleMode = async () => {
     if (mode === "edit") {
       setIsLoading(true)
-      await check()
+      await check(text)
     } else {
       setMode("edit")
     }
   }
 
   const handleReplace = async (index: number, newWord: string) => {
-    const updatedWords = [...words]
-    updatedWords[index].text = newWord
-    updatedWords[index].logprob = 0
-    updatedWords[index].replacements = []
+    console.log("handleReplace", index, newWord)
+    const updatedWords = words.map((w, i) => {
+      if (i === index) {
+        return { text: newWord, logprob: 0, replacements: [] }
+      }
+      return w
+    })
     setWords(updatedWords)
-    setText(updatedWords.map(w => w.text).join(""))
-    await check()
+    const newText = updatedWords.map(w => w.text).join("")
+    setText(newText)
+    await check(newText)
   }
 
   let result

frontend/src/components/WordChip.tsx

@@ -36,6 +36,7 @@ export const WordChip = ({
   }
 
   const handleReplacement = (event: React.ChangeEvent<HTMLSelectElement>) => {
+    console.log("handleReplacement", event.target.value)
     const newWord = event.target.value
     onReplace(newWord)
     setIsExpanded(false);
@@ -58,11 +59,12 @@ export const WordChip = ({
             position: "absolute",
             top: "100%",
             left: 0,
-            zIndex: 1000
+            zIndex: 1000,
+            overflowY: "auto"
           }}
-          size={replacements.length}
-          autoFocus
+          size={replacements.length + 1}
         >
+          <option key="original" hidden>{word}</option>
           {replacements.map((r, i) => (
             <option key={i} value={r}>{r}</option>
           ))}

main.py

@@ -1,15 +1,34 @@
 from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
+from functools import lru_cache
 
-from models import CheckResponse
+from models import ApiWord, CheckResponse
 from completions import check_text, load_model
 
 app = FastAPI()
 
-model, tokenizer, device = load_model()
+# model, tokenizer, device = load_model()
+
+def rep(i):
+    if i == 3:
+        return -10, [" jumped", " leaps"]
+    if i == 5:
+        return -10, [" calm"]
+    if i == 7:
+        return -10, [" dog", " cat", " bird", " fish"]
+    return -3, []
+
+@lru_cache(maxsize=100)
+def cached_check_text(text: str):
+    # return check_text(text, model, tokenizer, device)
+    result = []
+    for i, w in enumerate(text.split()):
+        logprob, replacements = rep(i)
+        result.append(ApiWord(text=f" {w}", logprob=logprob, replacements=replacements))
+    return result
 
 @app.get("/check", response_model=CheckResponse)
 def check(text: str):
-    return CheckResponse(text=text, words=check_text(text, model, tokenizer, device))
+    return CheckResponse(text=text, words=cached_check_text(text))
 
 app.mount("/", StaticFiles(directory="frontend/public", html=True))