From 34b8ffe5d7873c0ce90a113df5ef50c35e84765d Mon Sep 17 00:00:00 2001
From: Elizabeth Cray <elizabeth.cray@afs.com>
Date: Wed, 16 Oct 2024 09:46:04 -0400
Subject: [PATCH] Fix typo and add a couple more options

---
 meatgrinder.py | 28 +++++++++++++++++++++-------
 1 file changed, 21 insertions(+), 7 deletions(-)

diff --git a/meatgrinder.py b/meatgrinder.py
index c04c4d5..f14a061 100644
--- a/meatgrinder.py
+++ b/meatgrinder.py
@@ -7,6 +7,8 @@ import summarize
 parser = argparse.ArgumentParser(description='Transcribe and summarize meeting recordings')
 parser.add_argument('input', type=str, help='Path to the input media')
 parser.add_argument('--output', type=str, help='Path to the output file, will not print result to stdout if provided')
+parser.add_argument('--save-transcription', type=str, help='Path to save the transcription to')
+parser.add_argument('--skip-summary', action='store_true', help='Do not summarize')
 parser.add_argument('--force', action='store_true', help='Overwrite existing output file without asking')
 args = parser.parse_args()
 
@@ -14,20 +16,32 @@ if not args.input:
     print("Please provide an input file")
     exit(1)
 
-if args.output and os.path.isFile(args.output) and not args.force:
+if args.output and args.skip_summary:
+    print("Cannot save output file without summarizing, ignoring --output")
+
+if args.output and os.path.isfile(args.output) and not args.force and not args.skip_summary:
     print("Output file already exists and will be overwritten")
     if input("Continue? [y/N] ").lower() != "y":
         exit(1)
 
+if args.save_transcription:
+    if os.path.isfile(args.save_transcription) and not args.force:
+        print("Transcription file already exists and will be overwritten")
+        if input("Continue? [y/N] ").lower() != "y":
+            exit(1)
 
 audio_file = media_convert.process(args.input)
 
 transcription = transcribe.process(audio_file)
 
-summary = summarize.process(transcription)
+if args.save_transcription:
+    with open(args.save_transcription, 'w') as f:
+        f.write(transcription)
 
-if args.output:
-    with open(args.output, 'w') as f:
-        f.write(summary)
-else:
-    print(summary)
+if not args.skip_summary:
+    summary = summarize.process(transcription)
+    if args.output:
+        with open(args.output, 'w') as f:
+            f.write(summary)
+    else:
+        print(summary)