Update app.py

app.py

@@ -1,55 +1,77 @@
 import gradio as gr
 import pandas as pd
-import matplotlib.pyplot as plt
+from vega_datasets import data
 
-def update_dropdowns(file):
-    if file is None:
+def update_dropdowns(csv_file):
+    if csv_file is None:
         return gr.Dropdown(choices=[]), gr.Dropdown(choices=[])
-    df = pd.read_csv(file)
-    columns = list(df.columns)
-    return gr.Dropdown(choices=columns), gr.Dropdown(choices=columns)
+    df = pd.read_csv(csv_file.name)
+    numeric_cols = df.select_dtypes(include=['number']).columns.tolist()
+    return gr.Dropdown(choices=numeric_cols), gr.Dropdown(choices=numeric_cols)
 
-def create_scatter_plot(file, x_col, y_col):
-    if not file or not x_col or not y_col:
-        return None
-    
-    df = pd.read_csv(file)
-    
-    plt.figure(figsize=(10, 6))
-    plt.scatter(df[x_col], df[y_col])
-    plt.title(f"{x_col} vs {y_col}")
-    plt.xlabel(x_col)
-    plt.ylabel(y_col)
-    plt.grid(True)
+def plot_selection(selection, df):
+    if selection is None or df is None:
+        return pd.DataFrame()
+    return df.iloc[[i["index"] for i in selection]]
+
+def create_scatter_plot(csv_file, x_col, y_col):
+    if csv_file is None or x_col is None or y_col is None:
+        return None, None
     
-    plt.savefig("scatter_plot.png")
-    return "scatter_plot.png"
+    df = pd.read_csv(csv_file.name)
+    plot = gr.ScatterPlot(
+        value=df,
+        x=x_col,
+        y=y_col,
+        title="Custom CSV Scatter Plot",
+        x_title=x_col,
+        y_title=y_col,
+        tooltip=df.columns.tolist(),
+        color_legend_title="Values",
+        interactive=True,
+        brush_radius=5  # Lasso/polygon seçim için
+    )
+    return plot, df
 
 with gr.Blocks() as app:
-    gr.Markdown("# CSV Veri Görselleştirme Aracı")
+    gr.Markdown("## 📊 Custom CSV Scatter Plot with Selection")
     
     with gr.Row():
-        csv_file = gr.File(label="CSV Dosyası Yükle", type="filepath")
+        csv_upload = gr.UploadButton(label="Upload CSV", file_types=[".csv"])
+        x_axis = gr.Dropdown(label="X Axis Column")
+        y_axis = gr.Dropdown(label="Y Axis Column")
     
     with gr.Row():
-        x_axis = gr.Dropdown(label="X Ekseni Parametresi", interactive=True)
-        y_axis = gr.Dropdown(label="Y Ekseni Parametresi", interactive=True)
-    
-    plot_button = gr.Button("Grafiği Oluştur")
+        plot = gr.ScatterPlot(interactive=True, brush_radius=5)
+        selected_data = gr.DataFrame(label="Selected Points")
     
-    with gr.Row():
-        plot_output = gr.Image(label="Scatter Plot")
+    df_store = gr.State()
 
-    csv_file.change(
+    # CSV yüklendiğinde dropdown'ları güncelle
+    csv_upload.upload(
         fn=update_dropdowns,
-        inputs=csv_file,
+        inputs=csv_upload,
         outputs=[x_axis, y_axis]
     )
-    
-    plot_button.click(
+
+    # Eksenler seçildiğinde grafiği oluştur
+    x_axis.change(
+        fn=create_scatter_plot,
+        inputs=[csv_upload, x_axis, y_axis],
+        outputs=[plot, df_store]
+    )
+    y_axis.change(
         fn=create_scatter_plot,
-        inputs=[csv_file, x_axis, y_axis],
-        outputs=plot_output
+        inputs=[csv_upload, x_axis, y_axis],
+        outputs=[plot, df_store]
+    )
+
+    # Seçim yapıldığında verileri güncelle
+    plot.select(
+        fn=plot_selection,
+        inputs=df_store,
+        outputs=selected_data
     )
 
-app.launch()
+if __name__ == "__main__":
+    app.launch()