Update utils.py

utils.py

@@ -9,6 +9,8 @@ from fastapi import UploadFile
 from prompt_template import template_format_instructions, template
 from typing import List
 
+
+
 # Create a directory to store temporary files
 TEMP_DIR = "/temp_files"
 # if not os.path.exists(TEMP_DIR):
@@ -117,3 +119,29 @@ def generate_json_structured_resume(resume, chat_llm):
     result = chain.invoke({"text": resume})
 
     return result
+
+
+def delete_files_in_directory(directory):
+    """
+    Deletes all files in the specified directory.
+
+    Args:
+    directory (str): The path to the directory containing files to be deleted.
+
+    Returns:
+    None
+    """
+    # Check if the directory exists
+    if not os.path.exists(directory):
+        print(f"Directory '{directory}' does not exist.")
+        return
+
+    # Get a list of all files in the directory
+    files = os.listdir(directory)
+
+    # Iterate over each file and delete it
+    for file in files:
+        file_path = os.path.join(directory, file)
+        if os.path.isfile(file_path):
+            os.remove(file_path)
+            print(f"Deleted file: {file_path}")