From 55f79343d94fe81f792a96a9c8b0dec772d868cc Mon Sep 17 00:00:00 2001
From: Fam Zheng <fam@euphon.net>
Date: Sat, 24 Jan 2026 12:21:16 +0000
Subject: [PATCH] ai: return model_name

---
 .gitignore           | 1 +
 emblem5/ai/server.py | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index e94022a..c336858 100644
--- a/.gitignore
+++ b/.gitignore
@@ -7,3 +7,4 @@ build
 /api/db.sqlite3
 /emblemscanner-release
 models
+emblem5/ai/data/
diff --git a/emblem5/ai/server.py b/emblem5/ai/server.py
index 104c230..676039e 100755
--- a/emblem5/ai/server.py
+++ b/emblem5/ai/server.py
@@ -83,7 +83,8 @@ def find_best_frame(files):
 def do_qr_verify():
     start_time = time.time()
     fd = flask.request.form
-    model_path, model_md5 = download_model(fd.get('model', default_model))
+    model_name = fd.get('model', default_model)
+    model_path, model_md5 = download_model(model_name)
     model, transforms = load_model(model_path)
 
     frame_image, clarities = find_best_frame(flask.request.files)
@@ -110,6 +111,7 @@ def do_qr_verify():
             "probabilities": ', '.join([f'{v:.2%}' for k, v in probabilities]),
             "clarities": clarities,
             "model_md5": model_md5,
+            "model_name": model_name,
         }
     }