diff --git a/main.py b/main.py
index a6644a4..2fd4242 100644
--- a/main.py
+++ b/main.py
@@ -127,7 +127,7 @@ def generate_river_czml(river_df, runoff_df, level=4, start_time=None):
 
     # 为每条河段创建一个entity
     for idx in river_df.index:
-        print(idx)
+        river_index = river_df.loc[idx, "Index"]
         coords = mapping(river_df.loc[idx, "geometry"])["coordinates"]
         # new_coords = [coord for point in coords for coord in point]
         # print(new_coords)
@@ -150,7 +150,7 @@ def generate_river_czml(river_df, runoff_df, level=4, start_time=None):
 
         # 创建河段实体
         entity = {
-            "id": f"river_{idx}",
+            "id": f"{river_index}",
             "name": f"River Segment {idx}",
             "polyline": {
                 "positions": {