Merge pull request #3 from selfsigned/main

Add InvokeAI Management

app.js

@@ -65,6 +65,26 @@ app.get('/start_comfyui', (req, res) => {
     res.send('Started ComfyUI successfully!');
 });
 
+app.get('/stop_invokeai', (req, res) => {
+    exec('scripts/stop_invokeai.sh &', (error, stdout, stderr) => {
+        if (error) {
+            console.error(`Error stopping InvokeAI: ${error}`);
+        }
+    });
+
+    res.send('Stopped InvokeAI successfully!');
+});
+
+app.get('/start_invokeai', (req, res) => {
+    exec('scripts/start_invokeai.sh &', (error, stdout, stderr) => {
+        if (error) {
+            console.error(`Error starting InvokeAI: ${error}`);
+        }
+    });
+
+    res.send('Started InvokeAI successfully!');
+});
+
 app.listen(PORT, '0.0.0.0', () => {
     console.log(`Server is running on http://0.0.0.0:${PORT}`);
 });

public/config.json

@@ -2,6 +2,7 @@
   "applications": [
     "A1111",
     "Kohya_ss",
-    "ComfyUI"
+    "ComfyUI",
+    "InvokeAI"
   ]
 }

public/index.html

@@ -41,6 +41,13 @@ <h1 class="text-center">RunPod Application Manager</h1>
                     <button id="startComfyUI" class="btn btn-success">Start</button>
                 </td>
             </tr>
+            <tr id="rowInvokeAI">
+                <td>InvokeAI</td>
+                <td>
+                    <button id="stopInvokeAI" class="btn btn-danger">Stop</button>
+                    <button id="startInvokeAI" class="btn btn-success">Start</button>
+                </td>
+            </tr>
             </tbody>
         </table>
     </div>

public/script.js

@@ -45,4 +45,16 @@ $(document).ready(function() {
             alert(data);
         });
     });
+
+    $('#stopInvokeAI').click(function () {
+        $.get('/stop_invokeai', function (data) {
+            alert(data);
+        });
+    });
+
+    $('#startInvokeAI').click(function () {
+        $.get('/start_invokeai', function (data) {
+            alert(data);
+        });
+    });
 });

scripts/start_invokeai.sh

@@ -0,0 +1,2 @@
+#!/usr/bin/env bash
+cd /workspace/InvokeAI && source venv/bin/activate && nohup invokeai-web > /workspace/logs/invokeai.log 2>&1 &

scripts/stop_invokeai.sh

@@ -0,0 +1,2 @@
+#!/usr/bin/env bash
+fuser -k 9091/tcp