Update app.py

app.py

@@ -12,26 +12,57 @@ 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"])
+
 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)
+    with st.expander("Context Details"):
+        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"]:
@@ -44,7 +75,6 @@ def display_dict(data):
     st.write(data["output"]["perturbed_explanation"])
 
 
-
 # 主程序
 def main():
     st.title("Perturb For Table")