Update app.py

app.py

@@ -6,24 +6,24 @@ from data_loader import read_dask_data, read_polars_data, read_another_dask_data
 # Function to handle the data loading and processing logic
 def load_and_process_data(dataset_choice: str, num_rows: int) -> Dict[str, Any]:
     try:
-        # Load and process datasets based on user choice
-        if dataset_choice == "Dask Data":
-            data = read_dask_data()
-            processed_data = data.head(num_rows)
-        elif dataset_choice == "Polars Data":
-            data = read_polars_data()
-            processed_data = data.head(num_rows)
-        elif dataset_choice == "Another Dask Data":
-            data = read_another_dask_data()
-            processed_data = data.head(num_rows)
-        else:
+        # Load datasets based on user choice
+        dataset_mapping = {
+            "Dask Data": read_dask_data,
+            "Polars Data": read_polars_data,
+            "Another Dask Data": read_another_dask_data
+        }
+
+        # Fetch the chosen dataset using the mapping
+        data_loader = dataset_mapping.get(dataset_choice)
+        if not data_loader:
             return {"error": "Invalid dataset choice."}
 
-        # Optionally, include more complex data processing here
+        # Process data to show the specified number of rows
+        data = data_loader()
+        processed_data = data.head(num_rows)
 
-        # Return the processed data
         return {
-            "Processed Data": processed_data
+            "Processed Data": processed_data.to_dict()  # Convert DataFrame to dictionary for JSON serialization
         }
     except Exception as e:
         # Log the exception
@@ -32,9 +32,14 @@ def load_and_process_data(dataset_choice: str, num_rows: int) -> Dict[str, Any]:
 
 # Create a Gradio interface with more features
 def create_interface():
-    # Interface inputs
-    dataset_choice = gr.inputs.Dropdown(["Dask Data", "Polars Data", "Another Dask Data"], label="Select Dataset")
-    num_rows = gr.inputs.Slider(minimum=1, maximum=100, default=5, label="Number of Rows to Display")
+    # Interface inputs (updated for Gradio v5.x)
+    dataset_choice = gr.Dropdown(
+        choices=["Dask Data", "Polars Data", "Another Dask Data"], 
+        label="Select Dataset"
+    )
+    num_rows = gr.Slider(
+        minimum=1, maximum=100, default=5, label="Number of Rows to Display"
+    )
 
     # Interface definition
     demo = gr.Interface(