Update app.py

app.py

@@ -122,7 +122,20 @@ async def mergeStreams(request: Request):
 
         aggregated_df_b, log_dict = aggregate_data_optimized(df_A.copy(), df_B.copy(), 'uuid', 'graph_geometry_id')
 
-        aggregated_df_b_noGeo = aggregated_df_b.drop(columns=['@geometry', "@Geometry"]).fillna("NA")
+
+        # additional cleanups, remove geometry and fill  na
+        try:
+            aggregated_df_b_noGeo = aggregated_df_b.drop(columns=['@geometry']
+        except:
+            pass
+        try:
+            aggregated_df_b_noGeo = aggregated_df_b.drop(columns=['@Geometry']
+        except:
+            pass
+
+        aggregated_df_b_noGeo = aggregated_df_b.fillna("NA")
+
+        
         updateStreamAnalysisFast(
                 client = CLIENT,
                 stream_id ="ebcfc50abe",