solve lookup_compound

app.py

@@ -45,7 +45,12 @@ symspell.load_dictionary("symspell_dict_pythainlp.txt", term_index=0, count_inde
 
 # แก้คำผิด
 def correct_query_with_symspell(query: str) -> str:
-    suggestions = symspell.lookup_compound(query, Verbosity.CLOSEST, max_edit_distance=2)
+    # ถ้า query มีคำเดียว ใช้ lookup ปกติ
+    if len(query.strip().split()) == 1:
+        suggestions = symspell.lookup(query, Verbosity.CLOSEST, max_edit_distance=2)
+    else:
+        suggestions = symspell.lookup_compound(query, 2)
+    
     if suggestions:
         return suggestions[0].term
     return query