From e0b6c3b5a891228d9d61766d2710c6bcc5cdf2fa Mon Sep 17 00:00:00 2001
From: z3st <z3st@modelhub.org.cn>
Date: Fri, 24 Jul 2026 19:03:13 +0800
Subject: [PATCH] feat: use file-based SQLite database for persistent failure
 tracking

---
 Dockerfile | 2 ++
 main.py    | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 9d210a0..bca03b4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,6 +4,8 @@ ENV PYTHONUNBUFFERED=1
 
 WORKDIR /app
 
+RUN mkdir -p /app/data
+
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
diff --git a/main.py b/main.py
index bf5b240..6be1087 100644
--- a/main.py
+++ b/main.py
@@ -81,7 +81,8 @@ db_conn = None
 
 def init_db():
     global db_conn
-    db_conn = sqlite3.connect(':memory:', check_same_thread=False)
+    os.makedirs('/app/data', exist_ok=True)
+    db_conn = sqlite3.connect('/app/data/submit_history.db', check_same_thread=False)
     db_conn.execute('''CREATE TABLE IF NOT EXISTS submitted (
         model_id TEXT, gpu TEXT, task_id TEXT, status TEXT,
         submitted_at TEXT, checked_at TEXT,