Update App_Function_Libraries/Local_File_Processing_Lib.py

App_Function_Libraries/Local_File_Processing_Lib.py

@@ -1,90 +1,90 @@
-# Local_File_Processing_Lib.py
-#########################################
-# Local File Processing and File Path Handling Library
-# This library is used to handle processing local filepaths and URLs.
-# It checks for the OS, the availability of the GPU, and the availability of the ffmpeg executable.
-# If the GPU is available, it asks the user if they would like to use it for processing.
-# If ffmpeg is not found, it asks the user if they would like to download it.
-# The script will exit if the user chooses not to download ffmpeg.
-####
-
-####################
-# Function List
-#
-# 1. read_paths_from_file(file_path)
-# 2. process_path(path)
-# 3. process_local_file(file_path)
-# 4. read_paths_from_file(file_path: str) -> List[str]
-#
-####################
-
-# Import necessary libraries
-# Import Local
-from App_Function_Libraries.Audio_Transcription_Lib import convert_to_wav
-from App_Function_Libraries.Video_DL_Ingestion_Lib import *
-from App_Function_Libraries.Video_DL_Ingestion_Lib import get_youtube
-from App_Function_Libraries.Utils import normalize_title, create_download_directory
-
-#######################################################################################################################
-# Function Definitions
-#
-
-def read_paths_from_file(file_path):
-    """ Reads a file containing URLs or local file paths and returns them as a list. """
-    paths = []  # Initialize paths as an empty list
-    with open(file_path, 'r') as file:
-        paths = file.readlines()
-    return [path.strip() for path in paths]
-
-
-def process_path(path):
-    """ Decides whether the path is a URL or a local file and processes accordingly. """
-    if path.startswith('http'):
-        logging.debug("file is a URL")
-        # For YouTube URLs, modify to download and extract info
-        return get_youtube(path)
-    elif os.path.exists(path):
-        logging.debug("File is a path")
-        # For local files, define a function to handle them
-        return process_local_file(path)
-    else:
-        logging.error(f"Path does not exist: {path}")
-        return None
-
-
-# FIXME - ingest_text is not used, need to confirm.
-def process_local_file(file_path, ingest_text=False):
-    logging.info(f"Processing local file: {file_path}")
-    file_extension = os.path.splitext(file_path)[1].lower()
-
-    if os.path.isfile(file_path):
-        if file_path.lower().endswith('.txt'):
-            if ingest_text:
-                # Treat as content to be ingested
-                return os.path.dirname(file_path), {'title': os.path.basename(file_path)}, file_path
-            else:
-                # Treat as potential list of URLs
-                with open(file_path, 'r') as file:
-                    urls = file.read().splitlines()
-                return None, None, urls
-        elif file_path.lower().endswith(('.mp4', '.avi', '.mov', '.wav', '.mp3', '.m4a')):
-            # Handle video and audio files (existing code)
-            title = normalize_title(os.path.splitext(os.path.basename(file_path))[0])
-            info_dict = {'title': title}
-            logging.debug(f"Creating {title} directory...")
-            download_path = create_download_directory(title)
-            logging.debug(f"Converting '{title}' to an audio file (wav).")
-            audio_file = convert_to_wav(file_path)
-            logging.debug(f"'{title}' successfully converted to an audio file (wav).")
-            return download_path, info_dict, audio_file
-    else:
-        logging.error(f"File not found: {file_path}")
-        return None, None, None
-
-
-
-
-
-#
-#
+# Local_File_Processing_Lib.py
+#########################################
+# Local File Processing and File Path Handling Library
+# This library is used to handle processing local filepaths and URLs.
+# It checks for the OS, the availability of the GPU, and the availability of the ffmpeg executable.
+# If the GPU is available, it asks the user if they would like to use it for processing.
+# If ffmpeg is not found, it asks the user if they would like to download it.
+# The script will exit if the user chooses not to download ffmpeg.
+####
+
+####################
+# Function List
+#
+# 1. read_paths_from_file(file_path)
+# 2. process_path(path)
+# 3. process_local_file(file_path)
+# 4. read_paths_from_file(file_path: str) -> List[str]
+#
+####################
+
+# Import necessary libraries
+# Import Local
+from App_Function_Libraries.Audio_Transcription_Lib import convert_to_wav
+from App_Function_Libraries.Video_DL_Ingestion_Lib import *
+from App_Function_Libraries.Video_DL_Ingestion_Lib import get_youtube
+from App_Function_Libraries.Utils import normalize_title, create_download_directory
+
+#######################################################################################################################
+# Function Definitions
+#
+
+def read_paths_from_file(file_path):
+    """ Reads a file containing URLs or local file paths and returns them as a list. """
+    paths = []  # Initialize paths as an empty list
+    with open(file_path, 'r') as file:
+        paths = file.readlines()
+    return [path.strip() for path in paths]
+
+
+def process_path(path):
+    """ Decides whether the path is a URL or a local file and processes accordingly. """
+    if path.startswith('http'):
+        logging.debug("file is a URL")
+        # For YouTube URLs, modify to download and extract info
+        return get_youtube(path)
+    elif os.path.exists(path):
+        logging.debug("File is a path")
+        # For local files, define a function to handle them
+        return process_local_file(path)
+    else:
+        logging.error(f"Path does not exist: {path}")
+        return None
+
+
+# FIXME - ingest_text is not used, need to confirm.
+def process_local_file(file_path, ingest_text=False):
+    logging.info(f"Processing local file: {file_path}")
+    file_extension = os.path.splitext(file_path)[1].lower()
+
+    if os.path.isfile(file_path):
+        if file_path.lower().endswith('.txt'):
+            if ingest_text:
+                # Treat as content to be ingested
+                return os.path.dirname(file_path), {'title': os.path.basename(file_path)}, file_path
+            else:
+                # Treat as potential list of URLs
+                with open(file_path, 'r') as file:
+                    urls = file.read().splitlines()
+                return None, None, urls
+        elif file_path.lower().endswith(('.mp4', '.avi', '.mov', '.wav', '.mp3', '.m4a')):
+            # Handle video and audio files (existing code)
+            title = normalize_title(os.path.splitext(os.path.basename(file_path))[0])
+            info_dict = {'title': title}
+            logging.debug(f"Creating {title} directory...")
+            download_path = create_download_directory(title)
+            logging.debug(f"Converting '{title}' to an audio file (wav).")
+            audio_file = convert_to_wav(file_path)
+            logging.debug(f"'{title}' successfully converted to an audio file (wav).")
+            return download_path, info_dict, audio_file
+    else:
+        logging.error(f"File not found: {file_path}")
+        return None, None, None
+
+
+
+
+
+#
+#
 #######################################################################################################################