Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import pandas as pd
-import numpy as np
 from shapely.geometry import Point, Polygon
 
 def load_csv(file):
@@ -12,37 +11,37 @@ def update_dropdowns(df):
 def create_plot(df, x_col, y_col):
     if df is None or x_col is None or y_col is None:
         return None
-    return gr.ScatterPlot.update(
+    return gr.ScatterPlot(
         value=pd.DataFrame({x_col: df[x_col], y_col: df[y_col]}),
         x=x_col,
         y=y_col,
         title=f"{x_col} vs {y_col}",
         interactive=True,
-        tooltip=[x_col, y_col]
+        tooltip=[x_col, y_col],
+        elem_id="scatter"
     )
 
 def find_points_in_polygon(selected_data, df, x_col, y_col):
-    if not selected_data or not x_col or not y_col:
-        return pd.DataFrame()
-        
-    # Extract polygon points from selection
-    polygon_points = [(point["x"], point["y"]) for point in selected_data["points"]]
+    if not selected_data or df.empty or not x_col or not y_col:
+        return gr.DataFrame()
     
-    # Create Shapely objects
-    polygon = Polygon(polygon_points)
-    points = [Point(row[x_col], row[y_col]) for _, row in df.iterrows()]
-    
-    # Find points inside polygon
-    inside_indices = [i for i, point in enumerate(points) if polygon.contains(point)]
-    
-    return df.iloc[inside_indices]
+    try:
+        polygon_points = [(p["x"], p["y"]) for p in selected_data["points"]]
+        if len(polygon_points) < 3:
+            return gr.DataFrame()
+            
+        polygon = Polygon(polygon_points)
+        points = df[[x_col, y_col]].apply(lambda row: Point(row[x_col], row[y_col]), axis=1)
+        mask = points.apply(lambda p: polygon.contains(p))
+        return df[mask]
+    except Exception as e:
+        print(f"Error: {e}")
+        return gr.DataFrame()
 
 with gr.Blocks() as demo:
     gr.Markdown("## CSV Data Explorer with Polygon Selection")
     
-    # Store DataFrame and polygon points
-    df_state = gr.State()
-    polygon_state = gr.State([])
+    df_state = gr.State(pd.DataFrame())
     
     with gr.Row():
         csv_upload = gr.File(label="Upload CSV", file_types=[".csv"])
@@ -50,11 +49,10 @@ with gr.Blocks() as demo:
         y_col = gr.Dropdown(label="Y Column")
     
     plot = gr.ScatterPlot(interactive=True, label="Scatter Plot")
-    results = gr.DataFrame(label="Points in Polygon")
+    results = gr.DataFrame(label="Points in Polygon", render=False)
     
-    # Event handlers
     csv_upload.upload(
-        load_csv, 
+        load_csv,
         inputs=csv_upload,
         outputs=df_state
     ).then(