Added ability to set the caption extention in dataset viewer, captioner, and trainer so one dataset can have multiple caption styles in different files with different extensions. Added dataset caption template for a blank ideogram 4 formatted template.

ui/src/helpers/captionJobConfig.ts

@@ -21,6 +21,7 @@ export const defaultCaptionJobConfig: CaptionJobConfig = {
           extensions: ['mp3', 'wav', 'flac', 'ogg'],
           path_to_caption: '',
           recaption: false,
+          caption_extension: 'txt',
         },
       },
     ],

ui/src/helpers/datasetTemplates.ts

@@ -0,0 +1,21 @@
+const datasetTemplates: { [key: string]: string } = {
+    ideogram4: `
+{
+  "high_level_description": "",
+  "style_description": {
+    "aesthetics": "",
+    "lighting": "",
+    "photo": "",
+    "medium": "",
+    "color_palette": []
+  },
+  "compositional_deconstruction": {
+    "background": "",
+    "elements": [
+    ]
+  }
+}
+`
+};
+
+export default datasetTemplates;