File size: 3,739 Bytes
ed28876 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 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
#
#
####################################################################################################################### |