Update app.py

app.py

@@ -225,7 +225,22 @@ def extract_spatial_entities(text):
         final_doc.ents[i]._.rse_id = sent_rse_id[i]
 
     doc = final_doc
-    return doc.to_json()
+
+    ents_ext = []
+    for ent in doc.ents:
+        ents_ext.append({
+            "start": ent.start_char,
+            "end": ent.end_char,
+            "label": ent.label_,
+            "rse_id": ent._.rse_id  # ✅ 加入扩展字段
+        })
+
+    return {
+        "text": doc.text,
+        "ents": [{"start": ent.start_char, "end": ent.end_char, "label": ent.label_} for ent in doc.ents],
+        "tokens": [{"id": i, "start": token.idx, "end": token.idx + len(token)} for i, token in enumerate(doc)],
+        "ents_ext": ents_ext  # ✅ 添加扩展字段
+    }