Update app.py

app.py

@@ -5,53 +5,40 @@ import matplotlib.pyplot as plt
 from io import StringIO
 
 # Function to plot histogram
-def plot_histogram(file_contents, column):
+def plot_histogram(file_contents, column, ax=None):
     # Read the CSV file
     custom_df = pd.read_csv(StringIO(file_contents))
     
     # Plot histogram
-    plt.figure(figsize=(8, 6))
-    sns.histplot(custom_df[column])
-    plt.title(f'Histogram for {column}')
-    plt.xlabel(column)
-    plt.ylabel('Frequency')
-    return plt
+    sns.histplot(custom_df[column], ax=ax)
+    ax.set_title(f'Histogram for {column}')
+    ax.set_xlabel(column)
+    ax.set_ylabel('Frequency')
 
 # Function to plot scatter plot
-def plot_scatter(file_contents, x_axis, y_axis):
+def plot_scatter(file_contents, x_axis, y_axis, ax=None):
     # Read the CSV file
     custom_df = pd.read_csv(StringIO(file_contents))
     
     # Plot scatter plot
-    plt.figure(figsize=(8, 6))
-    sns.scatterplot(x=x_axis, y=y_axis, data=custom_df)
-    plt.title(f'Scatter Plot ({x_axis} vs {y_axis})')
-    plt.xlabel(x_axis)
-    plt.ylabel(y_axis)
-    return plt
-
-def layout_fn(inputs):
-    # Get the data from uploaded file
-    data = pd.read_csv(StringIO(inputs.file))
-
-    # Get selected columns from dropdown options
-    column_options = list(data.columns)
-    selected_column = inputs.text if inputs.text else None
-    selected_x_axis = inputs.text_1 if inputs.text_1 else None
-    selected_y_axis = inputs.text_2 if inputs.text_2 else None
+    sns.scatterplot(x=x_axis, y=y_axis, data=custom_df, ax=ax)
+    ax.set_title(f'Scatter Plot ({x_axis} vs {y_axis})')
+    ax.set_xlabel(x_axis)
+    ax.set_ylabel(y_axis)
 
+def layout_fn(file, text, text_1, text_2):
     # Create the figure with subplots
     fig, axes = plt.subplots(1, 2, figsize=(16, 6))
 
     # Check if data is uploaded and a column is selected for histogram
-    if inputs.file and selected_column:
-        plot_histogram(inputs.file.getvalue(), selected_column, ax=axes[0])
+    if file and text:
+        plot_histogram(file.getvalue(), text, ax=axes[0])
     else:
         axes[0].text(0.5, 0.5, "Upload a CSV and select a column", ha='center', va='center')
 
     # Check if data is uploaded and both x and y columns are selected for scatter plot
-    if inputs.file and selected_x_axis and selected_y_axis:
-        plot_scatter(inputs.file.getvalue(), selected_x_axis, selected_y_axis, ax=axes[1])
+    if file and text_1 and text_2:
+        plot_scatter(file.getvalue(), text_1, text_2, ax=axes[1])
     else:
         axes[1].text(0.5, 0.5, "Upload a CSV, select X and Y columns", ha='center', va='center')
 
@@ -60,32 +47,31 @@ def layout_fn(inputs):
     plt.tight_layout()
     return fig
 
-
 # Create the Gradio interface
 interface = gr.Interface(
     fn=layout_fn,
     inputs=[
-        gr.components.File(label="Upload CSV file"),
-        gr.components.Dropdown(label="Select Column (Histogram)", choices="infer"),
-        gr.components.Dropdown(label="Select X-axis (Scatter)", choices="infer"),
-        gr.components.Dropdown(label="Select Y-axis (Scatter)", choices="infer"),
+        gr.inputs.File(label="Upload CSV file"),
+        gr.inputs.Dropdown(label="Select Column (Histogram)", choices=[]),
+        gr.inputs.Dropdown(label="Select X-axis (Scatter)", choices=[]),
+        gr.inputs.Dropdown(label="Select Y-axis (Scatter)", choices=[]),
     ],
     outputs="plot",
     title="Data Visualization Tool",
     description="Upload a CSV file, select columns for histogram and scatter plots.",
 )
 
-
-@interface.update  # Use update method instead of component_args
-def update_choices(inputs):
-    if inputs.file:
-        data = pd.read_csv(StringIO(inputs.file.getvalue()))
+def update_choices(file):
+    if file:
+        data = pd.read_csv(StringIO(file.getvalue()))
         choices = list(data.columns)
-        interface.update(  # Update interface options using update
-            text=gr.components.Dropdown(choices=choices),
-            text_1=gr.components.Dropdown(choices=choices),
-            text_2=gr.components.Dropdown(choices=choices),
+        interface.set_config(
+            inputs=[
+                gr.inputs.File(label="Upload CSV file"),
+                gr.inputs.Dropdown(label="Select Column (Histogram)", choices=choices),
+                gr.inputs.Dropdown(label="Select X-axis (Scatter)", choices=choices),
+                gr.inputs.Dropdown(label="Select Y-axis (Scatter)", choices=choices),
+            ]
         )
 
-
-interface.launch(share=True,fn_change=update_choices)
+interface.run(share=True,fn_change=update_choices)