Hirad/Subtitle-Translator
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

create directories if not exists

Browse source
This commit is contained in:
Hirad 2025-02-27 15:32:09 +03:30
parent c643901237
commit 8525e753d4
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
gemini-st.py
View file

@ -4,6 +4,10 @@ from dotenv import load_dotenv
import gemini_srt_translator as gst import gemini_srt_translator as gst
def create_directory(directory_path):
os.makedirs(directory_path, exist_ok=True)
def translate(path, srt_file): def translate(path, srt_file):
load_dotenv() load_dotenv()
api_key = os.getenv('GEMINI_API_KEY') api_key = os.getenv('GEMINI_API_KEY')
@ -25,6 +29,7 @@ def translate(path, srt_file):
if __name__ == '__main__': if __name__ == '__main__':
home_dir = os.path.expanduser('~') home_dir = os.path.expanduser('~')
directory = os.path.join(home_dir, 'Documents', 'Subtitles') directory = os.path.join(home_dir, 'Documents', 'Subtitles')
create_directory(os.path.join(directory, 'translated'))
files = os.listdir(directory) files = os.listdir(directory)
srt_files = [f for f in files if f.endswith('.srt')] srt_files = [f for f in files if f.endswith('.srt')]
for file in srt_files: for file in srt_files: