From fde9fdddff1109613c4428cd5c5cd8f9933eeca4 Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Fri, 28 Mar 2025 11:46:05 +0800
Subject: [PATCH] Allow running with non working

---
 app/database/db.py |  2 ++
 main.py            | 11 ++++++++---
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/app/database/db.py b/app/database/db.py
index dceab503..c1951956 100644
--- a/app/database/db.py
+++ b/app/database/db.py
@@ -111,6 +111,8 @@ def init_db():
 
         session.commit()
 
+def can_create_session():
+    return Session is not None
 
 def create_session():
     return Session()
diff --git a/main.py b/main.py
index 0ce17192..80676ed9 100644
--- a/main.py
+++ b/main.py
@@ -138,7 +138,7 @@ import server
 from server import BinaryEventTypes
 import nodes
 import comfy.model_management
-from app.database.db import init_db
+from app.database.db import can_create_session, init_db
 from app.model_processor import model_processor
 
 def cuda_malloc_warning():
@@ -264,7 +264,11 @@ def start_comfyui(asyncio_loop=None):
 
     cuda_malloc_warning()
 
-    init_db()
+    try:
+        init_db()   
+    except Exception as e:
+        logging.error(f"Failed to initialize database. Please report this error as in future the database will be required: {e}")
+        
     prompt_server.add_routes()
     hijack_progress(prompt_server)
 
@@ -274,7 +278,8 @@ def start_comfyui(asyncio_loop=None):
         exit(0)
     
     # Scan for changed model files and update db
-    model_processor.run()
+    if can_create_session():
+        model_processor.run()
 
     os.makedirs(folder_paths.get_temp_directory(), exist_ok=True)
     call_on_start = None