Fix patch_frontend.py - handle edge cases in string search

patch_frontend.py

@@ -1,47 +1,110 @@
-"""Patch frontend ChatInput.tsx: Rename Claude→DeepSeek V3, remove Kimi, add new models."""
+"""Patch frontend ChatInput.tsx: Replace Claude with DeepSeek V3, remove Kimi, add models."""
 
 FILE = "/source/frontend/src/components/Chat/ChatInput.tsx"
 
 with open(FILE, "r") as f:
     content = f.read()
 
-# 1. Remove Kimi K2.6 entry entirely
-kimi_start = content.find("id: 'kimi-k2.6'")
-if kimi_start > -1:
-    # Find the opening { before it
-    brace_start = content.rfind("{", 0, kimi_start)
-    # Find the closing }, after it
-    brace_end = content.find("},", kimi_start) + 2
-    content = content[:brace_start] + content[brace_end:]
+# 1. Remove Kimi K2.6 entry
+kimi_idx = content.find("'kimi-k2.6'")
+if kimi_idx > -1:
+    # Walk back to find opening {
+    i = kimi_idx
+    brace_count = 0
+    while i > 0:
+        i -= 1
+        if content[i] == '{':
+            brace_count += 1
+            if brace_count == 1:
+                break
+    start = i
+    # Walk forward to find closing },
+    j = kimi_idx
+    brace_count = 0
+    while j < len(content):
+        if content[j] == '{':
+            brace_count += 1
+        elif content[j] == '}':
+            brace_count -= 1
+            if brace_count == 0:
+                break
+        j += 1
+    # Include the trailing comma and whitespace
+    end = j + 1
+    while end < len(content) and content[end] in ' ,\n':
+        end += 1
+    content = content[:start] + content[end:]
     print("✅ Removed Kimi K2.6")
+else:
+    print("⚠️ Kimi not found (may already be removed)")
 
-# 2. Replace Claude Opus entry with DeepSeek V3
-claude_start = content.find("id: 'claude-opus'")
-if claude_start > -1:
-    brace_start = content.rfind("{", 0, claude_start)
-    brace_end = content.find("},", claude_start) + 2
-    deepseek_entry = """  {
+# 2. Replace Claude entry with DeepSeek V3
+claude_idx = content.find("'claude-opus'")
+if claude_idx > -1:
+    i = claude_idx
+    brace_count = 0
+    while i > 0:
+        i -= 1
+        if content[i] == '{':
+            brace_count += 1
+            if brace_count == 1:
+                break
+    start = i
+    j = claude_idx
+    brace_count = 0
+    while j < len(content):
+        if content[j] == '{':
+            brace_count += 1
+        elif content[j] == '}':
+            brace_count -= 1
+            if brace_count == 0:
+                break
+        j += 1
+    end = j + 1
+    # Skip trailing comma
+    if end < len(content) and content[end] == ',':
+        end += 1
+
+    replacement = """{
     id: 'deepseek-v3',
     name: 'DeepSeek V3',
-    description: 'OpenRouter · Mặc định',
+    description: 'OpenRouter',
     modelPath: 'openai/deepseek/deepseek-chat-v3-0324',
     avatarUrl: 'https://huggingface.co/api/avatars/deepseek-ai',
     recommended: true,
   },"""
-    content = content[:brace_start] + deepseek_entry + content[brace_end:]
-    print("✅ Replaced Claude Opus with DeepSeek V3")
+    content = content[:start] + replacement + content[end:]
+    print("✅ Replaced Claude with DeepSeek V3")
+else:
+    print("⚠️ Claude entry not found")
 
-# 3. Add new models at end of array
-idx_deepseek = content.find("'deepseek-v4-pro'")
-if idx_deepseek == -1:
-    idx_deepseek = content.find("deepseek-ai/DeepSeek-V4")
-if idx_deepseek == -1:
-    # Try finding end of array another way
-    idx_deepseek = content.find("DEFAULT_MODEL_OPTIONS")
-    
-idx_end = content.find("];", idx_deepseek)
-if idx_end == -1:
-    print("❌ Cannot find end of array!")
+# 3. Add new models before the closing ]; of DEFAULT_MODEL_OPTIONS
+# Find the array end - search for "];\n" after DEFAULT_MODEL_OPTIONS
+arr_start = content.find("DEFAULT_MODEL_OPTIONS")
+if arr_start == -1:
+    print("❌ DEFAULT_MODEL_OPTIONS not found!")
+    import sys; sys.exit(1)
+
+# Find the closing ]; for this array
+idx = arr_start
+depth = 0
+found_open = False
+arr_end = -1
+while idx < len(content):
+    if content[idx] == '[' and not found_open:
+        found_open = True
+        depth = 1
+    elif content[idx] == '[':
+        depth += 1
+    elif content[idx] == ']':
+        depth -= 1
+        if depth == 0 and found_open:
+            arr_end = idx
+            break
+    idx += 1
+
+if arr_end == -1:
+    print("❌ Cannot find array end!")
     import sys; sys.exit(1)
 
 new_models = """  {
@@ -82,21 +145,16 @@ new_models = """  {
   },
 """
 
-content = content[:idx_end] + new_models + content[idx_end:]
-
-# 4. Fix FIRST_FREE_MODEL_PATH reference (was Kimi, now DeepSeek V3)
-content = content.replace(
-    "m.modelPath === FIRST_FREE_MODEL_PATH",
-    "m.modelPath === 'openai/deepseek/deepseek-chat-v3-0324'"
-)
+# Insert before the closing ]
+content = content[:arr_end] + new_models + content[arr_end:]
 
 with open(FILE, "w") as f:
     f.write(content)
 
 # Verify
-assert "kimi-k2.6" not in content, "Kimi still present!"
+assert "kimi-k2.6" not in content, "Kimi still in file!"
 assert "deepseek-v3" in content, "DeepSeek V3 missing!"
 assert "grok-4.3" in content, "Grok missing!"
 assert "qwen3.7-max" in content, "Qwen 3.7 missing!"
 assert "gemini-2.0-flash" in content, "Gemini missing!"
-print("\n✅ Frontend complete: DeepSeek V3 default, Kimi removed, all models added")
+print("\n✅ All done: DeepSeek V3 default, Kimi removed, 5 new models added")