Update CLI to remove LICENSE when creating new project & add more model options

Author: Shellishack

npm-packages/cli/source/components/commands/create.tsx

@@ -95,6 +95,12 @@ export default function Create({cli}: {cli: Result<Flags>}) {
 					fs.rmSync(githubDirPath, {recursive: true, force: true});
 				}
 
+				// Remove LICENSE file
+				const licenseFilePath = path.join(process.cwd(), name, 'LICENSE');
+				if (fs.existsSync(licenseFilePath)) {
+					fs.rmSync(licenseFilePath, {force: true});
+				}
+
 				setMessage(
 					<Box>
 						<Spinner type="dots" />

web/lib/modalities/image-gen/image-gen.ts

@@ -145,7 +145,11 @@ export function getImageGenModel(
       console.log({ prediction: prediction });
     }
 
-    const imgUrl = prediction.output[prediction.output.length - 1];
+    // Get the last output if output is an array, else if it is string get the output directly
+    const imgUrl =
+      typeof prediction.output === "string"
+        ? prediction.output
+        : prediction.output[prediction.output.length - 1];
 
     const arrayBuffer = await fetch(imgUrl).then((res) => res.arrayBuffer());
 

web/lib/modalities/image-gen/options.ts

@@ -13,6 +13,14 @@ export const imageGenProviderOptions: {
         model: "black-forest-labs/flux-schnell",
         isSupported: true,
       },
+      {
+        model: "black-forest-labs/flux-1.1-pro",
+        isSupported: true,
+      },
+      {
+        model: "black-forest-labs/flux-1.1-pro-ultra",
+        isSupported: true,
+      },
     ],
   },
 };

web/lib/modalities/video-gen/video-gen.ts

@@ -174,14 +174,18 @@ export function getVideoGenModel(
       console.log({ prediction: prediction });
     }
 
-    const imgUrl = prediction.output;
+    const videoUrl =
+      typeof prediction.output === "string"
+        ? prediction.output
+        : prediction.output[prediction.output.length - 1];
 
-    console.log("Video URL:", imgUrl);
 
-    const arrayBuffer = await fetch(imgUrl).then((res) => res.arrayBuffer());
+    console.log("Video URL:", videoUrl);
+
+    const arrayBuffer = await fetch(videoUrl).then((res) => res.arrayBuffer());
 
     return {
-      url: imgUrl,
+      url: videoUrl,
       arrayBuffer: arrayBuffer,
     };
   }