Update app.py

app.py

@@ -112,35 +112,51 @@ def reset_comp():
     return None, None, None
 
 # --------------------------
-# 🔥 diff 함수 (핵심)
+# 🔥 diff 함수
 # --------------------------
-def highlight_comp_only(base_text, comp_text):
+def highlight_diff(base_text, comp_text):
     if pd.isna(base_text): base_text = ""
     if pd.isna(comp_text): comp_text = ""
 
     base_words = str(base_text).split()
     comp_words = str(comp_text).split()
 
-    d = difflib.ndiff(base_words, comp_words)
+    d = list(difflib.ndiff(base_words, comp_words))
 
-    result = []
+    base_result = []
+    comp_result = []
 
-    for diff in d:
-        code = diff[0]
-        word = diff[2:]
+    i = 0
+    while i < len(d):
+        code = d[i][0]
+        word = d[i][2:]
 
         if code == ' ':
-            result.append(word)
+            base_result.append(word)
+            comp_result.append(word)
 
-        elif code == '+':  # 추가
-            result.append(
+        elif code == '-' and i + 1 < len(d) and d[i + 1][0] == '+':
+            old_word = word
+            new_word = d[i + 1][2:]
+
+            base_result.append(f"<span style='color:#ff4d4f; font-weight:600'>{old_word}</span>")
+            comp_result.append(f"<span style='color:#ff4d4f; font-weight:600'>{new_word}</span>")
+
+            i += 1
+
+        elif code == '-':
+            base_result.append(
+                f"<span style='color:#ff4d4f; text-decoration:line-through'>{word}</span>"
+            )
+
+        elif code == '+':
+            comp_result.append(
                 f"<span style='color:#2ecc71; font-weight:600'>{word}</span>"
             )
 
-        elif code == '-':
-            continue
+        i += 1
 
-    return " ".join(result)
+    return " ".join(base_result), " ".join(comp_result)
 
 # --------------------------
 # 데이터 조회
@@ -164,17 +180,10 @@ def display_data(standard, version, selection):
         cols = cols_info['name'].tolist()
         lower_cols = [c.lower().strip() for c in cols]
 
-        real_section = None
-        real_desc = None
-        real_ch = None
-        real_cat = None
+        real_ch, real_cat = None, None
 
         for c, lc in zip(cols, lower_cols):
-            if lc == "section":
-                real_section = c
-            elif lc == "description":
-                real_desc = c
-            elif lc == "chapter":
+            if lc == "chapter":
                 real_ch = c
             elif lc == "category":
                 real_cat = c
@@ -241,14 +250,14 @@ def unified_search(bs, bv, bc, cs, cv, cc):
             "description_y": f"Description_{cv}"
         })
 
-        # 🔥 diff 적용 (comp에만)
         base_col = f"Description_{bv}"
         comp_col = f"Description_{cv}"
 
+        # 🔥 diff 적용
         for idx, row in merged.iterrows():
-            base_text = row.get(base_col, "")
-            comp_text = row.get(comp_col, "")
-            merged.at[idx, comp_col] = highlight_comp_only(base_text, comp_text)
+            b, c = highlight_diff(row.get(base_col, ""), row.get(comp_col, ""))
+            merged.at[idx, base_col] = b
+            merged.at[idx, comp_col] = c
 
         merged = merged.sort_values(
             by="section",
@@ -281,8 +290,7 @@ with gr.Blocks() as demo:
             comp_category = gr.Dropdown(label="Category")
             comp_reset_btn = gr.Button("↺ 초기화")
 
-    with gr.Row():
-        search_btn = gr.Button("🔍 조회")
+    search_btn = gr.Button("🔍 조회")
 
     gr.Markdown("### 📊 비교 결과")
 
@@ -320,6 +328,26 @@ if __name__ == "__main__":
     font-family: 'Pretendard', sans-serif;
 }
 
+/* 테이블 레이아웃 */
+table {
+    table-layout: fixed;
+    width: 100%;
+}
+
+/* section 고정 */
+th:first-child, td:first-child {
+    width: 120px;
+    min-width: 120px;
+    max-width: 120px;
+}
+
+/* description 동일 분할 */
+th:nth-child(2), td:nth-child(2),
+th:nth-child(3), td:nth-child(3) {
+    width: calc((100% - 120px) / 2);
+}
+
+/* 헤더 고정 */
 thead th {
     position: sticky !important;
     top: 0;
@@ -328,15 +356,18 @@ thead th {
     border-bottom: 2px solid #ddd;
 }
 
+/* 스크롤 */
 .dataframe {
     max-height: 700px;
     overflow-y: auto;
 }
 
+/* 텍스트 */
 td {
     white-space: normal !important;
     word-break: break-word !important;
     line-height: 1.6;
+    padding: 10px;
 }
 """
     )