Update app.py

app.py

@@ -97,7 +97,8 @@ else:
     df_lu_filtered = df_lu[lu_columns].loc[mask_connected]
     df_lu_filtered.columns = [col.replace('lu+', '') for col in df_lu_filtered.columns]
     df_lu_filtered.columns = [col.replace('ASSETS+', '') for col in df_lu_filtered.columns]
-    
+
+    df_lu_filtered = df_lu_filtered.replace([np.inf, -np.inf], 10000).fillna(0)
     df_lu_filtered = df_lu_filtered.apply(pd.to_numeric, errors='coerce')
     df_lu_filtered = df_lu_filtered.astype(int)
     df_lu_filtered = df_lu_filtered.T.groupby(level=0).sum().T
@@ -126,10 +127,12 @@ def test(input_json):
     
         dfLanduses = pd.DataFrame(landuses).T
         dfLanduses = dfLanduses.apply(pd.to_numeric, errors='coerce')
+        dfLanduses = dfLanduses.replace([np.inf, -np.inf], 0).fillna(0)
         dfLanduses = dfLanduses.round(0).astype(int)
 
         dfMatrix = pd.DataFrame(matrix).T
         dfMatrix = dfMatrix.apply(pd.to_numeric, errors='coerce')
+        dfMatrix = dfMatrix.replace([np.inf, -np.inf], 10000).fillna(0)
         dfMatrix = dfMatrix.round(0).astype(int)
     else:
         dfLanduses = df_lu_filtered.copy()