Change source of AIS dataset

apps/home.py

@@ -6,7 +6,8 @@ import numpy as np
 import pandas as pd
 import streamlit as st
 from folium.plugins import MarkerCluster
-from streamlit_folium import folium_static 
+from streamlit_folium import folium_static
+from st_aggrid import GridOptionsBuilder, AgGrid, GridUpdateMode, DataReturnMode
 
 def app():
     st.write("# Welcome to Ship Detection Application! :satellite:")
@@ -24,7 +25,7 @@ def app():
 
         """
         )
-    ais = pd.read_csv("https://raw.githubusercontent.com/bills1912/marine-vessels-dataset/main/ais_data.csv")
+    ais = pd.read_csv("https://raw.githubusercontent.com/bills1912/marine-vessels-dataset/main/ais_dataV2.csv")
     ais_jakarta = ais[ais['Destination Port'] == 'JAKARTA']
     ais_list = ais_jakarta.values.tolist()
     f = folium.Figure(width=1000, height=500)
@@ -32,14 +33,35 @@ def app():
     ais_data = folium.FeatureGroup(name="marine_vessels")
     mCluster = MarkerCluster(name="Marine Vessels")
     for i in ais_list:
-        html = f"<h3>{i[1]}</h3> Vessel Type: {i[7]} </br> Destination Port: {i[2]} </br> Reported Destination: {i[4]} </br> Current Port: {i[5]}\
-                </br> Latitude: {i[9]} </br> Longitude: {i[10]}"
+        html = f"<h3>{i[1]}</h3> Vessel Type: {i[8]} </br> Destination Port: {i[2]} </br> Reported Destination: {i[4]} </br> Current Port: {i[5]}\
+                </br> Latitude: {i[10]} </br> Longitude: {i[11]}"
         iframe = folium.IFrame(html)
         popup = folium.Popup(iframe, min_width=250, max_width=300)
-        ais_data.add_child(mCluster.add_child(folium.Marker(location=[i[9], i[10]], popup=popup, icon=folium.Icon(color="black", icon="ship", prefix="fa"))))
+        ais_data.add_child(mCluster.add_child(folium.Marker(location=[i[10], i[11]], popup=popup, icon=folium.Icon(color="black", icon="ship", prefix="fa"))))
         jakarta_vessels.add_child(ais_data)
     folium_static(jakarta_vessels, width=1100, height=700)
     
+    # gb = GridOptionsBuilder.from_dataframe(ais)
+    # gb.configure_pagination(paginationAutoPageSize=True) #Add pagination
+    # gb.configure_selection('multiple', use_checkbox=False) #Enable multi-row selection
+    # gridOptions = gb.build()
+
+    # grid_response = AgGrid(
+    #     ais,
+    #     gridOptions=gridOptions,
+    #     data_return_mode='AS_INPUT', 
+    #     update_mode='MODEL_CHANGED', 
+    #     fit_columns_on_grid_load=False,
+    #     theme='dark', #Add theme color to the table
+    #     enable_enterprise_modules=True,
+    #     height=350, 
+    #     width='100%',
+    #     reload_data=True
+    # )
+
+    # data = grid_response['data']
+    # selected = grid_response['selected_rows'] 
+    
     st.markdown(
         """
         ## Explanation of The Ship Detection Result