Update core.py

core.py

@@ -1,4 +1,4 @@
-# /core.py
+# /core.py (Corrected)
 
 """
 Core business logic for the code generation application.
@@ -12,13 +12,15 @@ from typing import Dict, List, Optional, Tuple, Generator, Any
 from config import (
     HTML_SYSTEM_PROMPT, GENERIC_SYSTEM_PROMPT,
     HTML_SYSTEM_PROMPT_WITH_SEARCH, GENERIC_SYSTEM_PROMPT_WITH_SEARCH,
-    FOLLOW_UP_SYSTEM_PROMPT, SEARCH_START
+    FOLLOW_UP_SYSTEM_PROMPT
 )
 from services import llm_service, search_service
 from extractor import extract_text_from_file, extract_website_content
 from utils import (
-    history_to_messages, remove_code_block, process_image_for_model,
-    apply_search_replace
+    history_to_messages,
+    remove_code_block,
+    process_image_for_model,
+    apply_search_replace_changes  # <--- FIX: Corrected the function name here
 )
 
 # --- Type Definitions ---
@@ -90,18 +92,13 @@ def generate_code(
 
     for chunk in stream:
         content_stream += chunk
-        
+        processed_code = ""
+
         if is_follow_up:
-            # For follow-ups, we apply changes incrementally
-            # This logic assumes the model sends complete change blocks.
             last_html = history[-1][1] if history and history[-1][1] else ""
-            # Simple check to see if a full replacement block is present
-            if SEARCH_START in content_stream and content_stream.count(SEARCH_START) == content_stream.count(DIVIDER):
-                 modified_html = apply_search_replace(last_html, content_stream)
-                 processed_code = modified_html
-            else:
-                 # If not a full block, show the raw diff for user to see progress
-                 processed_code = f"Applying Changes:\n\n{content_stream}"
+            # <--- FIX: Use the corrected function name here
+            modified_html = apply_search_replace_changes(last_html, content_stream)
+            processed_code = modified_html
         else:
             processed_code = remove_code_block(content_stream)
         
@@ -111,12 +108,12 @@ def generate_code(
     final_content = content_stream
     if is_follow_up:
         last_html = history[-1][1] if history and history[-1][1] else ""
-        final_code = apply_search_replace(last_html, final_content)
+        # <--- FIX: And use the corrected function name here as well
+        final_code = apply_search_replace_changes(last_html, final_content)
     else:
         final_code = remove_code_block(final_content)
 
-    # Add the interaction to history
-    # For user content, use the original query, not the enhanced one, for cleaner history display
+    # Use the original query for history display, not the enhanced one.
     history.append((query, final_code))
     
     yield {"code_output": final_code, "history": history}