Update app.py

app.py

@@ -1,7 +1,5 @@
 import gradio as gr
 
-from get_vars import *
-'''
 import nltk
 import simplemma
 from nltk.tokenize import word_tokenize
@@ -25,7 +23,38 @@ def get_lists(file):
   return word_tokenized_text, word_tokenized_text_lower, sent_tokenized_text, sent_tokenized_text_lower
 
 words, words_lower, SENTENCES, SENTENCES_LOWER = get_lists(file)
-'''
+
+def search_engine_bot(target):
+
+  result = []
+  for i,sent in enumerate(sentences_lower):
+    if target.lower() in sent:
+      result.append(sentences[i])
+
+  if len(result) == 0:
+    return (f"Non ho trovato la parola '{target}' nei testi.\n", result)
+  
+  else:
+    return (f"""Ho trovato {len(result)} {"frasi" if len(result) > 1 else "frase"} in cui è presente la parola {target}.\n""", result)
+    
+def num_sentences(show=''):
+  try:
+    number = int(show)
+    return number
+  except:
+    number = ''
+    return number
+
+def show_results(result, number):
+    display = []
+    try:
+      for num,sent in enumerate(result[1][:int(number)]):
+        display.append(f"{num+1}: {sent}\n")
+      return display
+    except:
+      for num,sent in enumerate(result):
+        display.append(f"{num+1}: {sent}")
+      return display
 
 def search_engine(parola_da_cercare):
   sentences_lower = sentences_lower