validated ir.py

ir.py

@@ -27,3 +27,18 @@ class SearchEngine:
 		most_similar_idx = np.argmax(similarity_scores)
 
 		return self.prompts[most_similar_idx]
+
+
+if __name__ == "__main__":
+	storage = SearchEngine()
+
+	storage.add("I love programming")
+	storage.add("You need to graduate")
+	storage.add("Library is open")
+
+	input_prompt = "I enjoy coding"
+	most_similar_prompt = storage.search(input_prompt)
+
+	print(f"Input Prompt: {input_prompt}")
+	print(f"Most Similar Prompt: {most_similar_prompt}")
+