add parameter

get_spectrogram.py

@@ -62,7 +62,7 @@ if __name__ == "__main__":
     parser.add_argument('-n', '--columns_name', type=str, help='Name of the columns that contain the path of the .wav', default='Path')
     parser.add_argument('-i', '--input', type=str, choices=['file', 'folder'], help='Choose "file" if you have a .csv file or "folder" to export '
                                                                             'spectrogram from all the .wav of a folder', default='folder')
-    parser.add_argument('-f', '--file', type=str, help='Name of the file that contains the recording to print', default=None)
+    parser.add_argument('-f', '--file', type=str, help='Name of the file that contains the recording to print', default='None')
     parser.add_argument('--frames', type=int, help='Number of spectrogram per file', default=30)
     parser.add_argument('--duration', type=int, help='Duration for each spectrogram', default=8)
     parser.add_argument('--overlap', type=int, help='Overlap between 2 spectrograms', default=2)
@@ -84,6 +84,9 @@ if __name__ == "__main__":
         df = pd.read_csv(args.file, low_memory=False)
         df['Path'] = df[args.columns_name]
     elif args.input == 'folder':
+        if args.file != 'None'
+            df = pd.DataFrame(glob.glob(os.path.join(path_to_data, args.file), recursive=True), columns=['Path'])
+        else :
             df = pd.DataFrame(glob.glob(os.path.join(path_to_data, '*'), recursive=True), columns=['Path'])
 
     img_per_rec = [img_per_rec]*len(df.groupby('Path'))