add CLI interface

summarize_transcript.py

@@ -4,6 +4,7 @@ Script to summarize transcript using ERNIE-4.5-21B-A3B-PT-GGUF model with SYCL a
 """
 
 import os
+import argparse
 from llama_cpp import Llama
 from huggingface_hub import hf_hub_download
 
@@ -83,13 +84,21 @@ def stream_summarize_transcript(llm, transcript):
 
 
 def main():
+    parser = argparse.ArgumentParser(description="Summarize transcript using ERNIE-4.5-21B-A3B-PT-GGUF model.")
+    parser.add_argument("-i", "--input", type=str, default="./transcripts/short.txt", help="Path to the input transcript file (default: ./transcripts/short.txt)")
+    args = parser.parse_args()
+
     print("Loading ERNIE-4.5-21B-A3B-PT-GGUF model with SYCL acceleration...")
 
     # Load the model
     llm = load_model()
 
     # Read the transcript
-    transcript_path = "/home/luigi/tiny-scribe/transcripts/short.txt"
+    transcript_path = args.input
+    if not os.path.exists(transcript_path):
+        print(f"Error: Input file '{transcript_path}' not found.")
+        return
+
     transcript = read_transcript(transcript_path)
 
     print("\nOriginal Transcript (Preview):")
@@ -99,7 +108,7 @@ def main():
     summary = stream_summarize_transcript(llm, transcript)
 
     # Save summaries to files
-    with open("/home/luigi/tiny-scribe/summary.txt", 'w', encoding='utf-8') as f:
+    with open("summary.txt", 'w', encoding='utf-8') as f:
         f.write(summary)
 
     print("\nSummaries saved to summary.txt.")