now it saves the model in local directory #1333

Author: A-Akhil

NLP/Sentence_Similarity.py

@@ -1,12 +1,29 @@
+import os
 from sentence_transformers import SentenceTransformer, util
 
+MODEL_NAME = 'all-MiniLM-L6-v2'
+MODEL_FOLDER = 'model'
+
 def load_file(file_path):
     with open(file_path, 'r', encoding='utf-8') as file:
         return [line.strip() for line in file if line.strip()]
 
+def load_or_download_model():
+    model_path = os.path.join(MODEL_FOLDER, MODEL_NAME)
+    if os.path.exists(model_path):
+        print(f"Loading model from {model_path}")
+        return SentenceTransformer(model_path)
+    else:
+        print(f"Downloading model {MODEL_NAME}")
+        model = SentenceTransformer(MODEL_NAME)
+        os.makedirs(MODEL_FOLDER, exist_ok=True)
+        model.save(model_path)
+        print(f"Model saved to {model_path}")
+        return model
+
 def find_similar_sentences(query, file_path, top_n=5):
     # Load the pre-trained model
-    model = SentenceTransformer('all-MiniLM-L6-v2')
+    model = load_or_download_model()
 
     # Load and encode the sentences from the file
     sentences = load_file(file_path)
@@ -23,14 +40,27 @@ def find_similar_sentences(query, file_path, top_n=5):
 
     return top_results
 
+def validate_file_path(file_path):
+    if not file_path.endswith('.txt'):
+        file_path += '.txt'
+    if not os.path.exists(file_path):
+        raise FileNotFoundError(f"The file '{file_path}' does not exist.")
+    return file_path
+
 def main():
     print("Welcome to the Sentence Similarity Search Tool!")
 
     # Get user input for query
     query = input("Enter your query: ")
 
-    # Get user input for file path
-    file_path = input("Enter the path to your text file: ")
+    # Get user input for file path and validate it
+    while True:
+        file_path = input("Enter the path to your text file without extension: ")
+        try:
+            file_path = validate_file_path(file_path)
+            break
+        except FileNotFoundError as e:
+            print(f"Error: {str(e)} Please try again.")
 
     try:
         results = find_similar_sentences(query, file_path)
@@ -39,8 +69,6 @@ def main():
         for sentence, score in results:
             print(f"Similarity: {score:.4f}")
             print(f"Sentence: {sentence}\n")
-    except FileNotFoundError:
-        print(f"Error: The file '{file_path}' was not found. Please check the file path and try again.")
     except Exception as e:
         print(f"An error occurred: {str(e)}")