Use user chosen path for Python installation. (#87)

Comfy-Org · Oct 15, 2024 · 2faf6eb · 2faf6eb
1 parent a7dfeaf
commit 2faf6eb
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/src/main.ts b/src/main.ts
@@ -172,8 +172,7 @@ if (!gotTheLock) {
         });
       });
       await handleFirstTimeSetup();
-      const { userResourcesPath, appResourcesPath, pythonInstallPath, modelConfigPath, basePath } =
-        await determineResourcesPaths();
+      const { appResourcesPath, pythonInstallPath, modelConfigPath, basePath } = await determineResourcesPaths();
       SetupTray(mainWindow, basePath);
       port = await findAvailablePort(8000, 9999).catch((err) => {
         log.error(`ERROR: Failed to find available port: ${err}`);
@@ -760,7 +759,7 @@ function createDirIfNotExists(dirPath: string): void {
   }
 }
 
-async function createComfyConfigFile(userSettingsPath: string): Promise<void> {
+function createComfyConfigFile(userSettingsPath: string): void {
   const configContent: any = {
     'Comfy.ColorPalette': 'dark',
     'Comfy.NodeLibrary.Bookmarks': [],
@@ -776,7 +775,7 @@ async function createComfyConfigFile(userSettingsPath: string): Promise<void> {
   }
 
   try {
-    await fsPromises.writeFile(configFilePath, JSON.stringify(configContent, null, 2));
+    fs.writeFileSync(configFilePath, JSON.stringify(configContent, null, 2));
     log.info(`Created ComfyUI config file at: ${configFilePath}`);
   } catch (error) {
     log.error(`Failed to create ComfyUI config file: ${error}`);
@@ -861,17 +860,13 @@ async function determineResourcesPaths(): Promise<{
   }
 
   const defaultUserResourcesPath = getDefaultUserResourcesPath();
-  const defaultPythonInstallPath =
-    process.platform === 'win32'
-      ? path.join(path.dirname(path.dirname(app.getPath('userData'))), 'Local', 'comfyui_electron')
-      : app.getPath('userData');
 
   const appResourcePath = process.resourcesPath;
 
   // TODO(robinhuang): Look for extra models yaml file and use that as the userResourcesPath if it exists.
   return {
     userResourcesPath: defaultUserResourcesPath,
-    pythonInstallPath: defaultPythonInstallPath,
+    pythonInstallPath: basePath,
     appResourcesPath: appResourcePath,
     modelConfigPath: modelConfigPath,
     basePath: basePath,