return oov exception

app.py

@@ -19,6 +19,11 @@ def format_to_html(sim_output):
     rows += f"<tr><td>{word}</td><td style='text-align: right'>{round(sim, 4)}</td></tr>"
   return table.format(rows)
 
+def get_oov_word(error, words):
+  words = {"A": words[0], "B": words[1]}
+  key = str(error).split()[1].strip()[1:-1]
+  return words[key]
+
 def find_most_similar(word):
   sim_output = wv.most_similar(word)
   return format_to_html(sim_output)
@@ -26,14 +31,17 @@ def find_most_similar(word):
 def find_similarity(wordA, wordB):
   try:
     sim = wv.similarity(wordA, wordB)
-  except Exception:
-    return "[Error] OOV"
-  return max([float(sim), 0])
+  except Exception as e:
+    oov_word = get_oov_word(e, [wordA, wordB])
+    return f"[Error] {oov_word} is an OOV"
+  
+  return max([float(sim), 0.0])
 
 title = "Check Word Similarity With Literary Word2Vec"
 examples = ["སེམས་", "བླ་མ་", "སྟོབས་"]
 
-print(find_similarity(examples[0], examples[1]))
+assert isinstance(find_similarity(examples[0], examples[1]), float)
+assert isinstance(find_similarity("ར་", "B"), str)
 
 demo = gr.Interface(
   fn=find_similarity,