From b6b8be7b24e60d367d184a087ce93c719e774457 Mon Sep 17 00:00:00 2001
From: "Khoa.vo" <khoa.vo@MBP-C02GM4RQQ05N.local>
Date: Tue, 30 Dec 2025 21:25:54 +0700
Subject: [PATCH] fix: restore truncated logic in whisk_client.py (v4)

---
 push_registry.sh     |  2 +-
 templates/index.html |  2 +-
 whisk_client.py      | 21 +++++++++++----------
 3 files changed, 13 insertions(+), 12 deletions(-)

diff --git a/push_registry.sh b/push_registry.sh
index fc89d47..994effe 100755
--- a/push_registry.sh
+++ b/push_registry.sh
@@ -3,7 +3,7 @@
 # Configuration
 REGISTRY="git.khoavo.myds.me"
 IMAGE_NAME="vndangkhoa/apix"
-TAG="v3"
+TAG="v4"
 FULL_IMAGE="$REGISTRY/$IMAGE_NAME:$TAG"
 
 echo "=== Building Docker Image for Linux/AMD64 ==="
diff --git a/templates/index.html b/templates/index.html
index cba7248..7e372b2 100644
--- a/templates/index.html
+++ b/templates/index.html
@@ -19,7 +19,7 @@
         <aside class="sidebar">
             <div class="sidebar-header">
                 <div class="brand">
-                    <h1>aPix <span class="badge">v3.0 DEBUG</span></h1>
+                    <h1>aPix <span class="badge">v4.0 DEBUG</span></h1>
                 </div>
                 <div class="sidebar-header-actions">
                     <button type="button" class="toolbar-info-btn info-icon-btn" data-popup-target="help"
diff --git a/whisk_client.py b/whisk_client.py
index 06e61da..f8cf6d5 100644
--- a/whisk_client.py
+++ b/whisk_client.py
@@ -137,16 +137,17 @@ def generate_image_whisk(prompt, cookie_str, **kwargs):
 
     access_token = get_session_token(cookies)
     
-def load_template_favorites():
-    path = get_config_path('gallery_favorites.json') # Typo in original? No, it uses gallery_favorites for both? Let's check logic.
-    # Ah, original code had a separate logic or file? 
-    # Let me check the original code via view first to be safe.
-    # It seems there was no load_template_favorites definition in the previous view.
-    # I will replace the top of the file definitions if they exist there.
-    pass 
-
-# Retrying with correct strategy: replace the specific lines if I can find them.
-# I'll view the file top to locate these helpers first.
+    reference_image_path = kwargs.get('reference_image_path')
+    media_generation_id = None
+    if reference_image_path:
+        try:
+            media_generation_id = upload_reference_image(reference_image_path, cookies)
+        except Exception as e:
+            logger.error(f"Failed to upload reference image: {e}")
+            # Fallback to Text-to-Image? Or fail? 
+            # If user wants reference, we should probably fail or warn.
+            # For now, let's log and continue (media_generation_id will be None -> T2I)
+            pass
 
     aspect_ratio_map = {
         "1:1": "IMAGE_ASPECT_RATIO_SQUARE",