Update app.py

app.py

@@ -12,64 +12,27 @@ def load_json(file_name):
         return json.load(f)
 
 # 显示单个字典的信息
-# def display_dict(data):
-#     st.write("### 文件信息")
-#     st.write(f"**Path:** {data['path']}")
-#     st.write(f"**Table ID:** {data['table_id']}")
-#     st.write(f"**Section:** {data['section']}")
-#     st.write("### Table")
-#     st.markdown(data['table_html'], unsafe_allow_html=True)
-
-#     st.write("### Context")
-#     # 拼接 all_context 并高亮 target_context_ids 的句子
-#     all_context = data["all_context"]
-#     highlighted_context = ""
-#     for idx, sentence in enumerate(all_context):
-#         if idx == data["perturb_sentence_id"]:
-#             highlighted_context += f"<span style='color:red;'>{sentence}</span> "
-#         elif idx in data["target_context_ids"]:
-#             highlighted_context += f"**{sentence}** "
-#         else:
-#             highlighted_context += sentence + " "
-#     st.markdown(highlighted_context, unsafe_allow_html=True)
-
-#     st.write("### Selected Paragraphs")
-#     for paragraph in data["selected_paragraphs"]:
-#         st.write(paragraph)
-
-#     st.write("### Output")
-#     st.write("**Perturbed Statement:**")
-#     st.write(data["output"]["perturbed_statement"])
-#     st.write("**Perturbed Explanation:**")
-#     st.write(data["output"]["perturbed_explanation"])
-# 修改 Context 渲染逻辑，添加 CSS 控制布局
 def display_dict(data):
     st.write("### 文件信息")
-    st.write(f"**Path:** {data['path']} | **Table ID:** {data['table_id']} | **Section:** {data['section']}")
-
-    # 使用两个列来分隔内容
-    col1, col2 = st.columns(2)
+    st.write(f"**Path:** {data['path']}")
+    st.write(f"**Table ID:** {data['table_id']}")
+    st.write(f"**Section:** {data['section']}")
+    st.write("### Table")
+    st.markdown(data['table_html'], unsafe_allow_html=True)
+
+    st.write("### Context")
+    # 拼接 all_context 并高亮 target_context_ids 的句子
+    all_context = data["all_context"]
+    highlighted_context = ""
+    for idx, sentence in enumerate(all_context):
+        if idx == data["perturb_sentence_id"]:
+            highlighted_context += f"<span style='color:red;'>{sentence}</span> "
+        elif idx in data["target_context_ids"]:
+            highlighted_context += f"**{sentence}** "
+        else:
+            highlighted_context += sentence + " "
+    st.markdown(highlighted_context, unsafe_allow_html=True)
 
-    with col1:
-        st.write("### Table")
-        st.markdown(data['table_html'], unsafe_allow_html=True)
-
-    with col2:
-        st.write("### Context")
-        # 拼接 all_context，并高亮特定部分
-        all_context = data["all_context"]
-        highlighted_context = '<div class="context">'
-        for idx, sentence in enumerate(all_context):
-            if idx == data["perturb_sentence_id"]:
-                highlighted_context += f"<span style='color:red;'>{sentence}</span> "
-            elif idx in data["target_context_ids"]:
-                highlighted_context += f"<b>{sentence}</b> "
-            else:
-                highlighted_context += sentence + " "
-        highlighted_context += '</div>'
-        st.markdown(highlighted_context, unsafe_allow_html=True)
-
-    # 单独分离的内容
     st.write("### Selected Paragraphs")
     for paragraph in data["selected_paragraphs"]:
         st.write(paragraph)
@@ -81,6 +44,7 @@ def display_dict(data):
     st.write(data["output"]["perturbed_explanation"])
 
 
+
 # 主程序
 def main():
     st.title("Perturb For Table")