Update app.py

app.py

@@ -4,38 +4,57 @@ import matplotlib.pyplot as plt
 import fcsparser
 
 def parse_fcs(file):
-    # Parse the FCS file
-    meta, data = fcsparser.parse(file.name)
-    return pd.DataFrame(data)
+    try:
+        # Parse the FCS file
+        meta, data = fcsparser.parse(file.name)
+        df = pd.DataFrame(data)
+        
+        # Check if the DataFrame is empty
+        if df.empty:
+            raise ValueError("The parsed DataFrame is empty. Please check the .fcs file.")
+        
+        return df
+    except Exception as e:
+        raise gr.Error(f"Error parsing .fcs file: {str(e)}")
 
 def plot_fcs(data, x_axis, y_axis):
-    # Plot the data
-    plt.figure(figsize=(10, 6))
-    plt.scatter(data[x_axis], data[y_axis], alpha=0.5)
-    plt.xlabel(x_axis)
-    plt.ylabel(y_axis)
-    plt.title(f"{x_axis} vs {y_axis}")
-    return plt
+    try:
+        # Plot the data
+        plt.figure(figsize=(10, 6))
+        plt.scatter(data[x_axis], data[y_axis], alpha=0.5)
+        plt.xlabel(x_axis)
+        plt.ylabel(y_axis)
+        plt.title(f"{x_axis} vs {y_axis}")
+        return plt.gcf()  # Return the current figure
+    except Exception as e:
+        raise gr.Error(f"Error plotting data: {str(e)}")
 
 def process_fcs(file):
-    # Parse the FCS file and get the DataFrame
-    df = parse_fcs(file)
-    
-    # Get the column names for x and y axis choices
-    columns = df.columns.tolist()
-    
-    # Create a Gradio interface for selecting x and y axis
-    with gr.Blocks() as demo:
-        with gr.Row():
-            x_axis = gr.Dropdown(choices=columns, label="Select X-axis")
-            y_axis = gr.Dropdown(choices=columns, label="Select Y-axis")
-        plot = gr.Plot()
+    try:
+        # Parse the FCS file and get the DataFrame
+        df = parse_fcs(file)
+        
+        # Get the column names for x and y axis choices
+        columns = df.columns.tolist()
+        
+        # Check if columns are available
+        if not columns:
+            raise ValueError("No columns found in the DataFrame.")
+        
+        # Create a Gradio interface for selecting x and y axis
+        with gr.Blocks() as demo:
+            with gr.Row():
+                x_axis = gr.Dropdown(choices=columns, label="Select X-axis")
+                y_axis = gr.Dropdown(choices=columns, label="Select Y-axis")
+            plot = gr.Plot()
+            
+            # Update the plot when the dropdowns change
+            x_axis.change(lambda x, y, df: plot_fcs(df, x, y), inputs=[x_axis, y_axis, gr.State(df)], outputs=plot)
+            y_axis.change(lambda x, y, df: plot_fcs(df, x, y), inputs=[x_axis, y_axis, gr.State(df)], outputs=plot)
         
-        # Update the plot when the dropdowns change
-        x_axis.change(plot_fcs, inputs=[gr.State(df), x_axis, y_axis], outputs=plot)
-        y_axis.change(plot_fcs, inputs=[gr.State(df), x_axis, y_axis], outputs=plot)
-    
-    return demo
+        return demo
+    except Exception as e:
+        raise gr.Error(f"Error processing .fcs file: {str(e)}")
 
 # Gradio interface for uploading the FCS file
 with gr.Blocks() as demo: