Update app.py

app.py

@@ -1,3 +1,4 @@
+
 """
 AI HTML Editor — Flask Server
 Semua pemrosesan (AI call, find/change, fuzzy match) dilakukan di sisi server.
@@ -5,7 +6,7 @@ Job disimpan di memori dengan TTL 1 hari.
 """
 
 from flask import Flask, request, jsonify, Response
-import uuid, time, re, json, threading, requests
+import uuid, time, re, threading, requests
 
 app = Flask(__name__)
 
@@ -48,9 +49,17 @@ def update_job(jid: str, **kwargs):
             JOBS[jid].update(kwargs)
 
 # ─────────────────────────────────────────────
-#  ALGORITMA FIND & CHANGE  (ported dari JS)
+#  ALGORITMA FIND & CHANGE
 # ─────────────────────────────────────────────
 
+def decode_html_entities(text: str) -> str:
+    """Perbaiki jika AI mengembalikan HTML yang ter-escape secara tidak sengaja."""
+    return (text.replace('&', '&')
+                .replace('&lt;',  '<')
+                .replace('&gt;',  '>')
+                .replace('&quot;', '"')
+                .replace('&#39;', "'"))
+
 def levenshtein(a: str, b: str) -> int:
     la, lb = len(a), len(b)
     if la == 0: return lb
@@ -72,7 +81,6 @@ def calculate_similarity(s1: str, s2: str) -> float:
     return (len(longer) - levenshtein(longer, shorter)) / len(longer)
 
 def fuzzy_replace(source: str, search: str, replacement: str) -> dict:
-    """Sama persis dengan fuzzyReplace() di JS."""
     search_trimmed = search.strip()
     if not search_trimmed:
         return {"success": False, "result": source}
@@ -89,7 +97,7 @@ def fuzzy_replace(source: str, search: str, replacement: str) -> dict:
     except re.error:
         pass
 
-    # Level 2 — sliding window + Levenshtein (hanya jika < 3000 chars)
+    # Level 2 — sliding window + Levenshtein
     if len(search_trimmed) > 3000:
         return {"success": False, "result": source}
 
@@ -120,24 +128,18 @@ def fuzzy_replace(source: str, search: str, replacement: str) -> dict:
 
     return {"success": False, "result": source}
 
-
 def apply_find_change(code_base: str, ai_response: str) -> dict:
-    """
-    Parse respons AI dan terapkan semua blok <find>/<element>/<change>.
-    Return dict: {success, code, message, action_label}
-    """
     exec_match = re.search(r'<execution>([\s\S]*?)</execution>', ai_response, re.IGNORECASE)
     if not exec_match:
-        raise ValueError("Gunakan format tag <execution>.")
+        raise ValueError("AI Gagal menggunakan format tag <execution>.")
 
     exec_content = exec_match.group(1)
-
     respond_match = re.search(r'<responding>([\s\S]*?)</responding>', exec_content, re.IGNORECASE)
     respond_text  = respond_match.group(1).strip() if respond_match else "Kode berhasil diperbarui."
 
     find_blocks = re.findall(r'<find>([\s\S]*?)</find>', exec_content, re.IGNORECASE)
     if not find_blocks:
-        raise ValueError("Kamu wajib menggunakan tag <find> berisi <element> dan <change>.")
+        raise ValueError("AI tidak memberikan tag <find> yang berisi <element> dan <change>.")
 
     temp_code     = code_base
     success_count = 0
@@ -147,16 +149,13 @@ def apply_find_change(code_base: str, ai_response: str) -> dict:
         el_match  = re.search(r'<element>([\s\S]*?)</element>', block, re.IGNORECASE)
         ch_match  = re.search(r'<change>([\s\S]*?)</change>',   block, re.IGNORECASE)
         if not el_match or not ch_match:
-            raise ValueError(f"Blok <find> ke-{i+1} tidak memiliki <element> atau <change>.")
+            raise ValueError(f"Blok <find> ke-{i+1} formatnya salah (kehilangan element/change).")
 
-        old_code         = el_match.group(1)
+        # Decode entitas HTML yang mungkin tersalin sebagai &lt; dsb
+        old_code = decode_html_entities(el_match.group(1))
+        new_code = decode_html_entities(ch_match.group(1))
+        
         old_code_trimmed = old_code.strip()
-        new_code = (ch_match.group(1)
-                    .replace('&amp;', '&')
-                    .replace('&lt;',  '<')
-                    .replace('&gt;',  '>')
-                    .replace('&quot;', '"')
-                    .replace('&#39;', "'"))
 
         if old_code in temp_code:
             temp_code = temp_code.replace(old_code, new_code, 1)
@@ -172,15 +171,13 @@ def apply_find_change(code_base: str, ai_response: str) -> dict:
                 is_fuzzy = True
             else:
                 raise ValueError(
-                    f"Isi dari <element> pada <find> ke-{i+1}:\n{old_code}\n\n"
-                    "TIDAK DITEMUKAN di codebase meskipun fitur Similarity Fallback sudah aktif. "
-                    "Salin teks dengan lebih teliti!"
+                    f"Kode pada <element> ke-{i+1} TIDAK DITEMUKAN persis di Codebase.\n"
+                    f"Teks yang kamu kirim:\n{old_code[:100]}..."
                 )
 
     action_label = f"Diperbarui ({success_count} bagian){' ✨ Auto-Fix' if is_fuzzy else ''}"
     return {"code": temp_code, "message": respond_text, "action_label": action_label}
 
-
 # ─────────────────────────────────────────────
 #  AI CALL
 # ─────────────────────────────────────────────
@@ -212,7 +209,6 @@ def call_ai(prompt: str, model: str = DEFAULT_MODEL) -> str:
             return str(answer)
     raise RuntimeError(f"Format respons API tidak dikenali: {str(data)[:200]}")
 
-
 def build_system_prompt(code_base: str) -> str:
     return f"""Kamu adalah AI HTML Editor. Tugasmu HANYA mengubah bagian tertentu dari kode (Partial Update).
 Codebase saat ini:
@@ -220,14 +216,13 @@ Codebase saat ini:
 {code_base}
 ```
 
-ATURAN WAJIB:
-1. Gunakan blok <find> untuk setiap perubahan.
-2. <element> HARUS berisi teks/kode lama yang IDENTIK PERSIS dengan Codebase.
-3. <change> berisi kode baru penggantinya.
-4. Boleh memakai banyak blok <find> sekaligus.
-5. Balas HANYA dengan format <execution> seperti contoh di bawah. DILARANG KERAS menulis ulang seluruh kode!
+ATURAN WAJIB PENGGUNAAN BLOK <find>:
+1. <element> HARUS berisi teks/kode yang IDENTIK PERSIS dengan Codebase (termasuk spasi, tab, dan baris kosong). JANGAN perbaiki indentasi di dalam <element>!
+2. JANGAN memasukkan potongan kode yang terlalu panjang. Ambil sependek mungkin (1-4 baris) asalkan UNIK.
+3. <change> berisi kode baru yang akan menggantikan <element> tersebut.
+4. Balas HANYA dengan format <execution> seperti contoh. DILARANG menulis ulang seluruh codebase di luar tag.
 
-CONTOH JAWABAN:
+CONTOH JAWABAN YANG BENAR:
 <execution>
 <responding>Saya telah mengganti teks H1 dan warna tombol.</responding>
 <find>
@@ -240,24 +235,27 @@ CONTOH JAWABAN:
 </find>
 </execution>"""
 
-
 def run_ai_loop(jid: str, user_instruction: str, code_base: str,
                 loop: int = 0, error_feedback: str = None, max_loops: int = 10,
                 model: str = DEFAULT_MODEL):
-    """Rekursif di thread terpisah — sama dengan runAILoop() JS."""
+    """Rekursif self-healing AI."""
     if loop >= max_loops:
         update_job(jid,
             status="error",
-            error="Gagal merubah kode setelah 10x loop perbaikan. AI terus menerus gagal mencocokkan kode dengan Codebase. Coba beri instruksi yang lebih jelas.",
+            error="Gagal merubah kode setelah 10x loop perbaikan. AI terus menerus gagal mencocokkan kode dengan Codebase. Mohon salin sebagian kecil kode secara manual ke dalam chat jika ingin diubah.",
             loop_count=loop,
         )
         return
 
     try:
         prompt = build_system_prompt(code_base) + f"\n\nInstruksi User: {user_instruction}"
+        
         if error_feedback:
-            prompt += (f"\n\n[ERROR PADA JAWABANMU SEBELUMNYA]:\n{error_feedback}\n"
-                       "Perbaiki pencarian teksmu. Jika kau kesulitan, setidaknya salin elemen persis seperti aslinya.")
+            prompt += (f"\n\n[PENTING! ERROR PADA JAWABANMU SEBELUMNYA]:\n{error_feedback}\n\n"
+                       "INSTRUKSI PERBAIKAN:\n"
+                       "1. Kesalahan ini terjadi karena teks di <element> beda spasi/karakter dengan codebase.\n"
+                       "2. SOLUSI: Ambil potongan kode yang LEBIH PENDEK (1-2 baris saja) yang unik.\n"
+                       "3. Jangan mengubah spasi dan indentasi sedikitpun saat menyalin ke <element>!")
 
         ai_text = call_ai(prompt, model=model)
         result  = apply_find_change(code_base, ai_text)
@@ -269,7 +267,7 @@ def run_ai_loop(jid: str, user_instruction: str, code_base: str,
         )
 
     except Exception as exc:
-        # Self-healing: retry
+        # Self-healing: retry dengan mengirimkan error ke AI
         threading.Thread(
             target=run_ai_loop,
             args=(jid, user_instruction, code_base, loop + 1, str(exc), max_loops),
@@ -277,17 +275,12 @@ def run_ai_loop(jid: str, user_instruction: str, code_base: str,
             daemon=True,
         ).start()
 
-
 # ─────────────────────────────────────────────
 #  ROUTES
 # ─────────────────────────────────────────────
 
 @app.route("/api/chat", methods=["POST"])
 def api_chat():
-    """
-    Body: { instruction: str, code_base: str }
-    Return: { job_id: str }
-    """
     body = request.get_json(silent=True) or {}
     instruction = body.get("instruction", "").strip()
     code_base   = body.get("code_base", "")
@@ -308,10 +301,8 @@ def api_chat():
 
     return jsonify({"job_id": jid}), 202
 
-
 @app.route("/api/job/<jid>", methods=["GET"])
 def api_job_status(jid):
-    """Poll status job."""
     job = get_job(jid)
     if not job:
         return jsonify({"error": "Job tidak ditemukan atau sudah expired (> 1 hari)"}), 404
@@ -328,17 +319,14 @@ def api_job_status(jid):
 
     return jsonify(resp)
 
-
 @app.route("/api/job/<jid>", methods=["DELETE"])
 def api_delete_job(jid):
-    """Hapus job secara manual."""
     with _lock:
         if jid in JOBS:
             del JOBS[jid]
             return jsonify({"deleted": True})
     return jsonify({"error": "Job tidak ditemukan"}), 404
 
-
 # ─────────────────────────────────────────────
 #  UI  (embedded HTML)
 # ─────────────────────────────────────────────