Update app.py

app.py

@@ -30,23 +30,65 @@ def plot_scatter(file_contents, x_axis, y_axis):
     plt.ylabel(y_axis)
     return plt
 
-iface_histogram = gr.Interface(
-    fn=plot_histogram,
-    inputs=[gr.components.File(label="Upload CSV file"), gr.components.Text()],
-    outputs="plot",
-    title="Histogram Plotter",
-    description="Upload a CSV file and select a column to plot its histogram.",
-)
+def layout_fn(inputs, outputs):
+    # 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
+
+    # 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])
+    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])
+    else:
+        axes[1].text(0.5, 0.5, "Upload a CSV, select X and Y columns", ha='center', va='center')
 
-# Create the Gradio interface for scatter plot
-iface_scatter = gr.Interface(
-    fn=plot_scatter,
-    inputs=[gr.components.File(label="Upload CSV file"), gr.components.Text(), gr.components.Text()],
+    # Adjust layout
+    fig.suptitle("Data Visualization")
+    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"),
+    ],
     outputs="plot",
-    title="Scatter Plotter",
-    description="Upload a CSV file and select X and Y columns to plot a scatter plot.",
+    title="Data Visualization Tool",
+    description="Upload a CSV file, select columns for histogram and scatter plots.",
 )
 
-# Launch the Gradio interfaces
-iface_histogram.launch(share=True)
-iface_scatter.launch(share=True)
+# Set dropdown options dynamically based on uploaded data
+interface.component_args["text"] = {"choices": None}
+interface.component_args["text_1"] = {"choices": None}
+interface.component_args["text_2"] = {"choices": None}
+
+
+@interface.update(enabled=False)
+def update_choices(file):
+    if file:
+        data = pd.read_csv(StringIO(file.getvalue()))
+        choices = list(data.columns)
+        interface.component_args["text"] = {"choices": choices}
+        interface.component_args["text_1"] = {"choices": choices}
+        interface.component_args["text_2"] = {"choices": choices}
+
+
+interface.launch(share=True)