Update app.py

app.py

@@ -638,60 +638,74 @@ def run_tool(source_url, target_url, anchor_text, smart_rewrite, plain_text, sug
     # Process original anchor suggestion
     draft_html = res["best_sentence_with_anchor"]
 
-    # 1) Optional first-pass rewrite with language support
-    if smart_rewrite:
-        g = gpt_rewrite(draft_html, anchor_text, target_url, style="neutral", language=language_name)
-        final_html = g["sentence_html"]
+    # Check if anchor was already present in the article
+    anchor_was_present = res.get("anchor_was_present", False)
+    
+    # Only apply GPT rewriting if anchor wasn't already present
+    # If anchor is present, we just want to show where to add the link
+    if anchor_was_present:
+        # Anchor exists - just show where to add the link
+        final_output = to_plain_text(draft_html) if plain_text else draft_html
+        result = warn + f"✅ **Anchor text '{anchor_text}' found in article!**\n\n"
+        result += f"📍 Add link here:\n\n"
+        result += f"Original: {original_sentence}\n\n"
+        result += f"With link: {final_output}"
     else:
-        final_html = draft_html
-
-    # 2) QA/polish pass with language support
-    if not res.get("anchor_was_present", False):
+        # Anchor doesn't exist - need to add it to the sentence
+        # 1) Optional first-pass rewrite with language support
+        if smart_rewrite:
+            g = gpt_rewrite(draft_html, anchor_text, target_url, style="neutral", language=language_name)
+            final_html = g["sentence_html"]
+        else:
+            final_html = draft_html
+
+        # 2) QA/polish pass with language support
         polished = gpt_validate_and_polish(final_html, anchor_text, target_url, language=language_name)
         final_html = polished.get("sentence_html", final_html)
 
-    # 3) Optionally convert to plain text
-    final_output = to_plain_text(final_html) if plain_text else final_html
-
-    # Build the result - ONLY show Result 1 with the original anchor
-    result = warn + f"📍 Result 1:\n\n"
-    result += f"Original: {original_sentence}\n\n"
-    result += f"Suggested: {final_output}"
-    
-    # ONLY show alternative if:
-    # 1. suggest_alternative_anchor is True
-    # 2. The original anchor was NOT found in the article
-    # 3. We have an alternative suggestion
-    if suggest_alternative_anchor and not res.get("keyword_in_article", True) and res.get("alternative_anchor"):
-        alt_anchor = res["alternative_anchor"]
-        alt_sentence_original = res.get("alternative_sentence_original", "")
-        alt_sentence = res.get("alternative_sentence", "")
+        # 3) Optionally convert to plain text
+        final_output = to_plain_text(final_html) if plain_text else final_html
+
+        # Build the result for when anchor is NOT present
+        result = warn + f"⚠️ **Anchor text '{anchor_text}' not found in article**\n\n"
+        result += f"📍 Result 1 - Suggested placement:\n\n"
+        result += f"Original: {original_sentence}\n\n"
+        result += f"Suggested: {final_output}"
         
-        # Detect language for alternative sentence
-        if alt_sentence_original:
-            alt_detected_lang = detect_language(alt_sentence_original)
-            alt_language_name = get_language_name(alt_detected_lang)
-            
-            # Apply GPT rewriting to alternative as well
-            if smart_rewrite and alt_sentence:
-                alt_g = gpt_rewrite(alt_sentence, alt_anchor, target_url, style="neutral", language=alt_language_name)
-                alt_final = alt_g["sentence_html"]
-            else:
-                alt_final = alt_sentence
-            
-            # Polish if needed
-            if not res.get("alternative_exact_match", False):
-                alt_polished = gpt_validate_and_polish(alt_final, alt_anchor, target_url, language=alt_language_name)
-                alt_final = alt_polished.get("sentence_html", alt_final)
-            
-            alt_output = to_plain_text(alt_final) if plain_text else alt_final
+        # ONLY show alternative if:
+        # 1. suggest_alternative_anchor is True
+        # 2. The original anchor was NOT found in the article
+        # 3. We have an alternative suggestion
+        if suggest_alternative_anchor and not res.get("keyword_in_article", True) and res.get("alternative_anchor"):
+            alt_anchor = res["alternative_anchor"]
+            alt_sentence_original = res.get("alternative_sentence_original", "")
+            alt_sentence = res.get("alternative_sentence", "")
             
-            # Add alternative as Result 2
-            result += f"\n\n{'='*50}\n\n"
-            result += f"📍 Result 2:\n"
-            result += f"💡 Alternative anchor: '{alt_anchor}'\n\n"
-            result += f"Original: {alt_sentence_original}\n\n"
-            result += f"Suggested: {alt_output}"
+            # Detect language for alternative sentence
+            if alt_sentence_original:
+                alt_detected_lang = detect_language(alt_sentence_original)
+                alt_language_name = get_language_name(alt_detected_lang)
+                
+                # Apply GPT rewriting to alternative as well
+                if smart_rewrite and alt_sentence:
+                    alt_g = gpt_rewrite(alt_sentence, alt_anchor, target_url, style="neutral", language=alt_language_name)
+                    alt_final = alt_g["sentence_html"]
+                else:
+                    alt_final = alt_sentence
+                
+                # Polish if needed
+                if not res.get("alternative_exact_match", False):
+                    alt_polished = gpt_validate_and_polish(alt_final, alt_anchor, target_url, language=alt_language_name)
+                    alt_final = alt_polished.get("sentence_html", alt_final)
+                
+                alt_output = to_plain_text(alt_final) if plain_text else alt_final
+                
+                # Add alternative as Result 2
+                result += f"\n\n{'='*50}\n\n"
+                result += f"📍 Result 2 - Alternative from article:\n"
+                result += f"💡 Alternative anchor: '{alt_anchor}'\n\n"
+                result += f"Original: {alt_sentence_original}\n\n"
+                result += f"Suggested: {alt_output}"
     
     return result