Refactor print statements for consistency and clarity in video processing functions

app.py

@@ -21,17 +21,7 @@ def delete_files(files):
         except FileNotFoundError:
             print(f"File {file} not found for deletion.")
             pass
-    print("files deleted")
-
-def classify_input_format(input_string):
-    seconds = HHMMSS_to_sec(time_str=input_string)
-    if seconds is not None:
-        print("Successfully converted timestamps to seconds")
-        return seconds
-    else:    
-        # if the input format is neither seconds nor HH:MM:SS, return None
-        gr.Error("Input time stamp format not supported. Please enter a valid time in the format HH:MM:SS or seconds.")
-        return None
+    print("...files deleted")
 
 def set_output_file(input_file, output_format, folder, insert_string = 'trimmed'):
     # Set output file name and extension
@@ -39,9 +29,7 @@ def set_output_file(input_file, output_format, folder, insert_string = 'trimmed'
         os.makedirs(folder)
     file_name = f"{Path(input_file.name).stem.partition('.')[0]}_{insert_string}.{output_format}"
     output_file = os.path.join(folder, file_name)
-    print(f"Output folder: {folder}")
-    print(f"file_name: {file_name}")
-    print(f"Output file: {output_file}")
+    print(f"...set output file: {output_file}")
     return output_file
 
 def trim_video_helper(input_file, output_file, start_time, end_time):
@@ -50,42 +38,39 @@ def trim_video_helper(input_file, output_file, start_time, end_time):
         end_time = 300
     elif start_time and not end_time:
         end_time = 300 + HHMMSS_to_sec(time_str=start_time)
-    print("start time (s): ", start_time)
-    print("end time (s): ", end_time)
+    print("...start time (s): ", start_time)
+    print("...end time (s): ", end_time)
     # Trim the video
-    print("start trimming")
+    print("...start trimming")
     output_file = convert_and_trim_video(input_file.name, output_file, start_time, end_time)
-    print("finished trimming")
+    print("...finished trimming")
     return output_file
 
 def convert_video_helper(input_file, output_file, output_format):
     # convert video
-    print(f"convert_video_helper: converting {input_file} to {output_format}")
     output_file = convert_and_trim_video(input_file.name, output_file)
-    print("convert_video_helper: finished converting")
+    print("...finished converting")
     return output_file
 
 def convert_transcript_helper(input_transcript, output_transcript):
     # convert transcript
-    print("start converting transcript")
     table = xlsx_to_table(xl_file=input_transcript)
-    print("finished converting transcript to table")
+    print("...parsed transcript to table")
     output_file = table_to_ELAN_tsv(table, output_transcript)
-    print("finished converting transcript")
+    print("...finished converting transcript")
     return output_file
 
 
 
 def trim_video_vtr(input_file, output_format):
-    # trim video of 10-mintue session with a random selected start time
+    print(f"BEGIN TASK: trimming {input_file} to 10-minute interval with random start time")
+    # trim video to 10-min interval with a random selected start time
     try: 
         # randomly select start time
         start_time = random.randint(300, 900)
         end_time = start_time + 600 # since 10 minutes
-        print("start time (s): ", start_time)
-        print("end time (s): ", end_time)
         audio_base_name = input_file.name.split("/")[-1].split(".")[-2]
-        print("audio_base_name: ", audio_base_name)
+        print("...audio_base_name: ", audio_base_name)
         
         # set output file
         insert_string = f"start{start_time}_end{end_time}"
@@ -103,6 +88,7 @@ def trim_video_vtr(input_file, output_format):
         output_file = trim_video_helper(input_file, output_file, start_time, end_time)
 
         # delete threading
+        print("Done trimming. Deleting files...")
         path_to_delete = [input_file.name, output_file, log_file]
         threading.Thread(target=delete_files, args=([path_to_delete])).start()
         
@@ -113,6 +99,7 @@ def trim_video_vtr(input_file, output_format):
 
 
 def trim_video_wt(input_file, input_transcript, output_format, start_time, end_time):
+    print(f"BEGIN TASK: trimming {input_file} with transcript {input_transcript} from {start_time} to {end_time}")
     # trim video with transcript
     try:
         # set output file
@@ -128,7 +115,7 @@ def trim_video_wt(input_file, input_transcript, output_format, start_time, end_t
         output_transcript = convert_transcript_helper(path, output_transcript)
         # output_transcript = output_file
         # remove file after 10 minutes for security
-        print("start deleting files")
+        print("Done trimming. Deleting files...")
         path_to_delete = [input_file.name, input_transcript.name, output_file, output_transcript]
         threading.Thread(target=delete_files, args=([path_to_delete])).start()    
 
@@ -140,7 +127,6 @@ def trim_video_wt(input_file, input_transcript, output_format, start_time, end_t
 
 def sort_transcript_helper(input_transcript, output_transcript):
     # sort transcript
-    print("start sorting transcript")
     print("input_transcript: ", input_transcript)
     print("output_transcript: ", output_transcript)
     output_transcript = sort_transcript(input_transcript, output_transcript)
@@ -148,12 +134,14 @@ def sort_transcript_helper(input_transcript, output_transcript):
     return output_transcript
 
 def sort_transcript_wrapper(input_file):
+    print(f"\nBEGIN TASK: sorting transcript {input_file}")
     output_folder = f"{os.getcwd()}/results/"
     output_file_path = set_output_file(input_file, "tsv", output_folder, insert_string = 'sorted')
     output_file_path = sort_transcript_helper(input_file.name, output_file_path)
     return output_file_path
 
 def trim_video(input_file, output_format, start_time, end_time):
+    print(f"\nBEGIN TASK: trimming {input_file} from {start_time} to {end_time}")
     try:
         # Set output file
         output_folder = f"{os.getcwd()}/results/"
@@ -163,7 +151,7 @@ def trim_video(input_file, output_format, start_time, end_time):
         output_file = trim_video_helper(input_file, output_file, start_time, end_time)
 
         # Remove files after 10 minutes for security
-        print("start deleting files")
+        print("Done trimming. Deleting files...")
         path_to_delete = [input_file.name, output_file]
         threading.Thread(target=delete_files, args=([path_to_delete])).start()
 
@@ -173,17 +161,17 @@ def trim_video(input_file, output_format, start_time, end_time):
         return f"Error: {str(e)}"
 
 def convert_video(input_file, output_format):
+    print(f"\nBEGIN TASK: converting {input_file} to {output_format}")
     try:
         # Set output file
         output_folder = f"{os.getcwd()}/results/"
         output_file = set_output_file(input_file, output_format, output_folder, \
                                       insert_string = 'converted')
-        print(f"Begin converting to output format: {output_format}" )
         # Convert video
         output_file = convert_video_helper(input_file, output_file, output_format)
-        print(f"created output file: {output_file}")
+        print(f"...created output file: {output_file}")
         # remove file after 10 minutes for security 
-        print("Done converting. start deleting files")
+        print("Done converting. Deleting files...")
         path_to_delete = [input_file.name, output_file]
         threading.Thread(target=delete_files, args=([path_to_delete])).start()