Add file flag

MultiSpeakerTranscriber.py

@@ -3,24 +3,11 @@
 import sys
 import argparse
 
-API_KEY = os.getenv("MULTISPEAKER_API")
-if API_KEY is None:
-	print("You are missing a value at $ASSEMBLY_AI_API. Set your API key and run again")
-	sys.exit(3)
-
-# Take in an argument of the file
-parser = argparse.ArgumentParser("MultispeakerTranscribe")
-parser.add_argument("audio", help="The filename of an audio file to be transcribed.")
-args = parser.parse_args()
-
-transcribe_file(args.audio)
-
-
 # Methods
 def transcribe_file(file):
 	# Chop off extension from file
 	filename = os.path.splitext(file)[0]
-
+	
 	try:
 		f = open("./Transcripts/" + filename + ".txt", "x")
 	except FileExistsError:
@@ -37,7 +24,7 @@ def transcribe_file(file):
 	print("Proceeding to transcribe the file.")
 
 	transcript = transcriber.transcribe(
-		"./Interviews/"+file,
+		file,
 		config=config
 	)
 
@@ -52,3 +39,15 @@ def transcribe_file(file):
 	# Delete the transcript so the only copy is local.
 	print("Proceeding to transcript id: " + transcript.id)
 	transcript.delete_by_id(transcript.id)
+
+API_KEY = os.getenv("MULTISPEAKER_API")
+if API_KEY is None:
+	print("You are missing a value at $ASSEMBLY_AI_API. Set your API key and run again")
+	sys.exit(3)
+
+# Take in an argument of the file
+parser = argparse.ArgumentParser("MultispeakerTranscribe")
+parser.add_argument("--file", dest='audio', help="The path of the audio file to be transcribed.")
+args = parser.parse_args()
+
+transcribe_file(args.audio)

README.md

@@ -8,7 +8,7 @@ You'll need to run `pip3 install assemblyai` (should work for macOS users) or `p
 
 When running MultispeakerTranscription, it currently expects to have an `Interviews` directory and a `Transcripts` directory in the same directory where it is being run. For example, if you have downloaded this file to `~/Downloads`, there should be `~/Downloads/Interviews` and `~/Downloads/Transcripts`. Then you can run this with `python3 MultiSpeakerTranscriber.py "<filename>"`. You only need to provide the file name. I recommend including quotation marks around to avoid splitting.
 
-Example: `python3 MultiSpeakerTranscriber.py "Montgomery Ward 1873-11-02.m4a"`
+Example: `python3 MultiSpeakerTranscriber.py --file "Montgomery Ward 1873-11-02.m4a"`
 
 ### What is this $SHELL
 *Note: This guide is currently Unix-centric*