highlight_diff_texts

app.py

@@ -758,27 +758,40 @@ def generate_correct_grammatical_spelling_errors(model, sys_content, eng_level,
 
 
 def highlight_diff_texts(highlight_list, text):
-    # Convert DataFrame to JSON string
-    highlight_list_json = highlight_list.to_json()
-
-    # Print the JSON string to see its structure
-    print("=======highlight_list_json=======")
-    print(highlight_list_json)
-
-    # Parse JSON string back to dictionary
-    highlight_list_dict = json.loads(highlight_list_json)
-
-    # Extract suggestions from the parsed JSON
-    suggestions = [highlight_list_dict['建議'][str(i)] for i in range(len(highlight_list_dict['建議']))]
-
-    # Initialize the HTML for text
-    text_html = f"<p>{text}</p>"
-
-    # Replace each suggestion in text with highlighted version
-    for suggestion in suggestions:
-        text_html = text_html.replace(suggestion, f'<span style="color:red;">{suggestion}</span>')
-
-    return text_html
+    """
+    將文本中需要高亮的部分用紅色標記出來
+    
+    Args:
+        highlight_list: 包含需要高亮的文字列表的 DataFrame
+        text: 原始文本
+    
+    Returns:
+        str: 包含高亮標記的 HTML 文本
+    """
+    try:
+        # 將 DataFrame 轉換為字典
+        highlight_dict = highlight_list.to_dict('records')
+        
+        # 初始化 HTML 文本
+        html_text = text
+        
+        # 遍歷每個需要高亮的建議
+        for item in highlight_dict:
+            if '原文' in item and item['原文']:
+                original = str(item['原文'])
+                # 使用更簡單的替換方式
+                html_text = html_text.replace(
+                    original,
+                    f'<span style="color:red;">{original}</span>'
+                )
+        
+        # 包裝在段落標籤中
+        return f"<p>{html_text}</p>"
+        
+    except Exception as e:
+        print(f"Error in highlight_diff_texts: {str(e)}")
+        # 發生錯誤時返回原始文本
+        return f"<p>{text}</p>"
 
 def update_paragraph_correct_grammatical_spelling_errors_input(paragraph):
     return paragraph