Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import pandas as pd
+import plotly.graph_objects as go
 from shapely.geometry import Point, Polygon
-import numpy as np
 
 def load_csv(file):
     return pd.read_csv(file.name)
@@ -9,50 +9,69 @@ def load_csv(file):
 def update_dropdowns(df):
     return gr.Dropdown(choices=df.columns.tolist()), gr.Dropdown(choices=df.columns.tolist())
 
-def create_plot(df, x_col, y_col):
+def create_plot(df, x_col, y_col, polygon_points=None):
     if df is None or x_col is None or y_col is None:
         return None
-    return gr.ScatterPlot(
-        value=pd.DataFrame({x_col: df[x_col], y_col: df[y_col]}),
-        x=x_col,
-        y=y_col,
+    
+    fig = go.Figure()
+    
+    # Ana scatter plot
+    fig.add_trace(go.Scatter(
+        x=df[x_col],
+        y=df[y_col],
+        mode='markers',
+        name='Data Points',
+        marker=dict(color='blue')
+    ))
+    
+    # Polygon çizgileri
+    if polygon_points and len(polygon_points) >= 2:
+        fig.add_trace(go.Scatter(
+            x=[p[0] for p in polygon_points],
+            y=[p[1] for p in polygon_points],
+            mode='lines+markers',
+            line=dict(color='red', width=2),
+            marker=dict(color='red', size=8),
+            name='Polygon'
+        ))
+    
+    fig.update_layout(
         title=f"{x_col} vs {y_col}",
-        interactive=True,
-        tooltip=[x_col, y_col],
-        color="blue"
+        dragmode='drawclosedpath',
+        showlegend=False
     )
+    
+    return fig
 
-def find_points_in_polygon(selected_data, df, x_col, y_col):
-    if not selected_data or df.empty or not x_col or not y_col:
-        return pd.DataFrame()
+def handle_click(click_data, df, x_col, y_col, polygon_points):
+    if not click_data or not df.empty:
+        return [], pd.DataFrame()
     
-    # Get selected polygon coordinates from lasso tool
-    polygon_points = [(p["x"], p["y"]) for p in selected_data["points"]]
-    if len(polygon_points) < 3:
-        return pd.DataFrame()
+    # Tıklanan noktayı al
+    point = (click_data['points'][0]['x'], click_data['points'][0]['y'])
+    new_points = polygon_points + [point]
     
-    # Create polygon and check containment
-    polygon = Polygon(polygon_points)
-    mask = df.apply(lambda row: polygon.contains(Point(row[x_col], row[y_col])), axis=1)
+    # Polygon tamamlandıysa içinde kalan noktaları bul
+    if len(new_points) >= 3 and new_points[0] == new_points[-1]:
+        polygon = Polygon(new_points)
+        mask = df.apply(lambda row: polygon.contains(Point(row[x_col], row[y_col])), axis=1)
+        return [], df[mask]
     
-    return df[mask]
+    return new_points, pd.DataFrame()
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Interactive CSV Explorer with Polygon Selection")
+    gr.Markdown("## Interactive Polygon Selection Tool")
     
     df_state = gr.State(pd.DataFrame())
+    polygon_state = gr.State([])
     
     with gr.Row():
-        csv_upload = gr.File(label="Upload CSV", file_types=[".csv"])
-        x_col = gr.Dropdown(label="X Column")
-        y_col = gr.Dropdown(label="Y Column")
+        csv_upload = gr.File(label="CSV Yükle", file_types=[".csv"])
+        x_col = gr.Dropdown(label="X Sütunu")
+        y_col = gr.Dropdown(label="Y Sütunu")
     
-    plot = gr.ScatterPlot(
-        interactive=True,
-        label="Scatter Plot",
-        show_download_button=True
-    )
-    results = gr.DataFrame(label="Points in Polygon")
+    plot = gr.Plot(label="Scatter Plot ve Polygon")
+    results = gr.DataFrame(label="Seçilen Veriler")
     
     csv_upload.upload(
         load_csv,
@@ -66,20 +85,20 @@ with gr.Blocks() as demo:
     
     x_col.change(
         create_plot,
-        inputs=[df_state, x_col, y_col],
+        inputs=[df_state, x_col, y_col, polygon_state],
         outputs=plot
     )
     
     y_col.change(
         create_plot,
-        inputs=[df_state, x_col, y_col],
+        inputs=[df_state, x_col, y_col, polygon_state],
         outputs=plot
     )
     
     plot.select(
-        find_points_in_polygon,
-        inputs=[df_state, x_col, y_col],
-        outputs=results
+        handle_click,
+        inputs=[df_state, x_col, y_col, polygon_state],
+        outputs=[polygon_state, results]
     )
 
 demo.launch()