diff --git a/lib.py b/lib.py index d4246e2..42b46e8 100644 --- a/lib.py +++ b/lib.py @@ -11,7 +11,7 @@ def read_files(path, output, filetypes=None): for root, dirs, files in os.walk(path): for file in files: if (file_type := file.split(".")[-1]) in filetypes: - output.append((file_type, os.path.join(path, file))) + output.append((file_type, os.path.join(path, file), file)) for folder in dirs: read_files(os.path.join(path, folder), output) diff --git a/main.py b/main.py index b18cfd4..0951d05 100644 --- a/main.py +++ b/main.py @@ -21,11 +21,14 @@ def retrieve_file_contents(path): contents.append({ "type": file[0], "path": file[1], + "filename": file[2], "content": content }) return contents -contents = retrieve_file_contents("C:\\SoftwareEng") +contents = retrieve_file_contents("C:\\Users\\afist\\Downloads\\SoftwareEng") + + print(json.dumps(contents, indent="\t")) \ No newline at end of file diff --git a/test_mp3parser.py b/test_mp3parser.py index c6d90b8..2d7b02f 100644 --- a/test_mp3parser.py +++ b/test_mp3parser.py @@ -5,7 +5,7 @@ import json model = whisper.load_model('tiny') -result = model.transcribe(str("Aufzeichnung.mp3"), language='de', verbose=True) +result = model.transcribe(str("C:\\Users\\afist\\Downloads\\Aufzeichnung.mp3"), language='de', verbose=True) # Dump the results to a JSON file with open('transcript.json', "w") as file: