create directories if not exists
This commit is contained in:
parent
c643901237
commit
8525e753d4
|
@ -4,6 +4,10 @@ from dotenv import load_dotenv
|
|||
import gemini_srt_translator as gst
|
||||
|
||||
|
||||
def create_directory(directory_path):
|
||||
os.makedirs(directory_path, exist_ok=True)
|
||||
|
||||
|
||||
def translate(path, srt_file):
|
||||
load_dotenv()
|
||||
api_key = os.getenv('GEMINI_API_KEY')
|
||||
|
@ -25,6 +29,7 @@ def translate(path, srt_file):
|
|||
if __name__ == '__main__':
|
||||
home_dir = os.path.expanduser('~')
|
||||
directory = os.path.join(home_dir, 'Documents', 'Subtitles')
|
||||
create_directory(os.path.join(directory, 'translated'))
|
||||
files = os.listdir(directory)
|
||||
srt_files = [f for f in files if f.endswith('.srt')]
|
||||
for file in srt_files:
|
||||
|
|
Loading…
Reference in a new issue