save current changes

src/index_prompt.ts

@@ -3,6 +3,12 @@ import { generateVideo } from './lib/video-generator';
 import { logger } from './lib/logger';
 import { Scene } from './types';
 import scenes from './scenes/space.json';
+import dotenv from 'dotenv';
+
+dotenv.config();
+
+const COMFY_BASE_URL = process.env.COMFY_BASE_URL;
+const COMFY_OUTPUT_DIR = process.env.COMFY_OUTPUT_DIR;
 
 interface ProcessedScene {
     scene: Scene;
@@ -10,6 +16,11 @@ interface ProcessedScene {
 }
 
 async function prepareImageForScene(scene: Scene): Promise<ProcessedScene | null> {
+    if (!COMFY_BASE_URL || !COMFY_OUTPUT_DIR) {
+        logger.error('COMFY_BASE_URL and COMFY_OUTPUT_DIR must be set in the environment variables.');
+        return null;
+    }
+
     try {
         const startTime = Date.now();
         logger.info(`--- Preparing image for scene: ${scene.idea} ---`);
@@ -19,6 +30,8 @@ async function prepareImageForScene(scene: Scene): Promise<ProcessedScene | null
         const generatedImagePath = await generateImage(
             scene.image_prompt,
             imageFileName,
+            COMFY_BASE_URL,
+            COMFY_OUTPUT_DIR,
             'flux',
             { width: 1280, height: 720 }
         );
@@ -35,6 +48,11 @@ async function prepareImageForScene(scene: Scene): Promise<ProcessedScene | null
 }
 
 async function generateVideoForScene(processedScene: ProcessedScene) {
+    if (!COMFY_BASE_URL || !COMFY_OUTPUT_DIR) {
+        logger.error('COMFY_BASE_URL and COMFY_OUTPUT_DIR must be set in the environment variables.');
+        return;
+    }
+
     const { scene, generatedImagePath } = processedScene;
     try {
         const startTime = Date.now();
@@ -45,6 +63,8 @@ async function generateVideoForScene(processedScene: ProcessedScene) {
             scene.video_prompt,
             generatedImagePath,
             videoFileName,
+            COMFY_BASE_URL,
+            COMFY_OUTPUT_DIR,
             { width: 1280, height: 720 }
         );
         const endTime = Date.now();