Update app.py

app.py

@@ -12,9 +12,15 @@ text = list(df["text"].values)
 
 def chitti(query):
     re_table = search(query, text)
-    with open(query[0:5]+'.txt', 'w') as f:
-        f.writelines(["Question: "+query,"Ans 1 : "+ re_table[0][0],"Ans 2 : "+ re_table[1][0]])
-    return [re_table[0][0],re_table[1][0]]
+    answers_re_table = [re_table[i][0] for i in range(0,5)]
+    sorted_indices = sorted(range(len(answers_re_table)), key=lambda k: len(answers_re_table[k]))
+    repeated_answers_indices =list()
+    for i in range(4):
+        if answers_re_table[sorted_indices[i]] in answers_re_table[sorted_indices[i+1]]:
+            repeated_answers_indices.append(sorted_indices[i])
+    for idx in repeated_answers_indices:
+        answers_re_table.pop(idx)
+    return [answers_re_table[0],answers_re_table[1]]
 
 demo = gr.Interface(
     fn=chitti,