update

app.py

@@ -179,22 +179,14 @@ def find_relations_with_entities(entities, entities_data):
     return relations
 
 
-def main():
-    """Main application"""
-    st.title("Medical Report Review System")
-
-    # Load data
-    if 'data' not in st.session_state:
-        st.session_state.data = load_data()
-
-    # Create two columns layout
+def setup_report_selection():
+    """Setup report selection columns and return selected report"""
     col1, col2 = st.columns(2)
 
     with col1:
         st.subheader("Reports to Review")
-        # Get unreviewed reports list
         unreviewed_reports = [
-            report_id for report_id, content in st.session_state.data.items()
+            report_id for report_id, content in st.session_state.reports_json.items()
             if 'reviewed' not in content
         ]
         selected_report = st.selectbox(
@@ -205,9 +197,8 @@ def main():
 
     with col2:
         st.subheader("Reviewed Reports")
-        # Get reviewed reports list
         reviewed_reports = [
-            report_id for report_id, content in st.session_state.data.items()
+            report_id for report_id, content in st.session_state.reports_json.items()
             if content.get('reviewed', False)
         ]
         st.selectbox(
@@ -216,59 +207,84 @@ def main():
             key="reviewed"
         )
 
-    if selected_report:
-        report_data = st.session_state.data[selected_report]
+    return selected_report
+
+
+def display_report_content(report_data):
+    """Display the report text content"""
+    st.subheader("Report Content:")
+    st.markdown(report_data['text'])
+
+
+def display_entities(report_text, entities):
+    """Setup and display entity annotation interface"""
+
+    st.subheader("Entity Annotation:")
+    selections = label_select(
+        body=report_text,
+        labels=list(set(e.labels[0] for e in entities)),
+        selections=entities,
+    )
+    st.write(selections)
+    return selections, entities
+
+
+def display_relationship_graph(entities, entities_data):
+    st.subheader("Entity Relationship Graph:")
+    relations = find_relations_with_entities(entities, entities_data)
+    create_graph(entities, relations)
+
+
+def handle_review_submission(selected_report, selections, entities_data):
+    """Handle the review submission process"""
+    if st.button("Mark as Reviewed"):
+        updated_entities = selection2entities(selections)
 
-        # Display report text
-        st.subheader("Report Content:")
-        st.markdown(report_data['text'])
+        for entity_id, entity in updated_entities.items():
+            if entity_id in entities_data:
+                entity['relations'] = entities_data[entity_id]['relations']
 
-        # Display entities and relations
+        st.session_state.reports_json[selected_report]['reviewed'] = {
+            'entities': updated_entities
+        }
+
+        save_data(st.session_state.reports_json)
+        st.success("Review status saved!")
+        st.rerun()
+
+
+def main():
+    """Main application"""
+    st.title("Medical Report Review System")
+
+    # Load data
+    if 'reports_json' not in st.session_state:
+        st.session_state.reports_json = load_data()
+
+    # Setup report selection
+    selected_report = setup_report_selection()
+
+    if selected_report:
+        report_data = st.session_state.reports_json[selected_report]
         entities_data = report_data['entities']
 
-        # Convert entities data to Selection objects
-        entities = entities2Selection(report_data['text'], entities_data)
+        # Display report content
+        display_report_content(report_data)
 
-        # Display annotation results
-        st.subheader("Entity Annotation:")
-        selections = label_select(
-            body=report_data['text'],
-            labels=list(set(e.labels[0] for e in entities)),
-            selections=entities,
-        )
-        # Display selections in text format
-        st.write(selections)
-
-        # Display relationship graph
-        st.subheader("Entity Relationship Graph:")
-
-        # Add update graph button
-        if st.button("Update Graph"):
-            # Update relations using current selections
-            relations = find_relations_with_entities(selections, entities_data)
-            create_graph(selections, relations)
-        else:
-            # Display original graph
-            relations = find_relations_with_entities(entities, entities_data)
-            create_graph(entities, relations)
-
-        # Add review functionality
-        if st.button("Mark as Reviewed"):
-            # Convert selections back to entities data
-            updated_entities = selection2entities(selections)
-
-            # Keep original relations
-            for entity_id, entity in updated_entities.items():
-                if entity_id in entities_data:
-                    entity['relations'] = entities_data[entity_id]['relations']
-
-            st.session_state.data[selected_report]['reviewed'] = {
-                'entities': updated_entities
-            }
-
-            save_data(st.session_state.data)
-            st.success("Review status saved!")
-            st.rerun()
+        # Setup entity annotation
+        selections = entities2Selection(report_data['text'], entities_data)
+        # Display entities
+        display_entities(report_data['text'], selections)
+
+        # Display NEN
+
+        # 使用当前选择或原始实体创建关系图
+        current_entities = selections if st.button(
+            "Update Graph") else selections
+        display_relationship_graph(current_entities, entities_data)
+
+        # Handle review submission
+        handle_review_submission(selected_report, selections, entities_data)
 
 
 if __name__ == "__main__":

test.py

@@ -0,0 +1,76 @@
+import streamlit as st
+
+# 存储标注信息
+if "annotations" not in st.session_state:
+    st.session_state.annotations = []
+
+# 添加标注
+
+
+def add_annotation(selected_text, label):
+    start_idx = text.find(selected_text)
+    if start_idx != -1:
+        end_idx = start_idx + len(selected_text)
+        st.session_state.annotations.append(
+            {"text": selected_text, "label": label, "start": start_idx, "end": end_idx})
+
+# 移除标注
+
+
+def remove_annotation(annotation):
+    st.session_state.annotations = [
+        ann for ann in st.session_state.annotations if ann != annotation
+    ]
+
+# 生成带按钮的高亮文本
+
+
+def render_highlighted_text(text, annotations):
+    annotated_text = []
+    last_idx = 0
+    for idx, ann in enumerate(sorted(annotations, key=lambda x: x["start"])):
+        # 添加非标注的部分
+        annotated_text.append(text[last_idx:ann["start"]])
+        # 添加标注的按钮
+        if st.button(f"{ann['text']} ({ann['label']})", key=f"annotation-{idx}"):
+            st.session_state.clicked_annotation = ann
+        last_idx = ann["end"]
+    # 添加剩余的部分
+    annotated_text.append(text[last_idx:])
+    return "".join(annotated_text)
+
+
+# 主界面
+st.title("文本标注工具")
+st.write("输入文本并对其中的部分内容进行标注。")
+
+# 输入文本
+text = st.text_area("输入文本：", height=200)
+
+# 文本选择与标注
+if text:
+    st.write("选中文字并进行标注：")
+    col1, col2 = st.columns([3, 1])
+
+    # 输入标注的标签
+    with col1:
+        selected_text = st.text_input("选中的文本：")
+    with col2:
+        label = st.text_input("标签：")
+        if st.button("添加标注"):
+            if selected_text and label and selected_text in text:
+                add_annotation(selected_text, label)
+            else:
+                st.warning("请选择有效的文本并输入标签！")
+
+    # 渲染高亮文本
+    st.markdown(render_highlighted_text(text, st.session_state.annotations))
+
+    # 点击标注的文本
+    if "clicked_annotation" in st.session_state:
+        clicked_annotation = st.session_state.clicked_annotation
+        st.write(
+            f"你点击了: **{clicked_annotation['text']}** (标签: {clicked_annotation['label']})")
+        if st.button("删除标注", key="delete"):
+            remove_annotation(clicked_annotation)
+            del st.session_state.clicked_annotation