Update app.py

app.py

@@ -1,12 +1,19 @@
 import gradio as gr
-import fcsparser
+import flowio
 import matplotlib.pyplot as plt
 import io
+import pandas as pd
 
 # Function to parse FCS file and extract data and metadata
 def parse_fcs(file):
-    meta, data = fcsparser.parse(file.name, reformat_meta=True)
-    return data, list(data.columns)
+    # Use flowio to parse the FCS file
+    fcs_data = flowio.FlowData(file.name)
+
+    # Extract channel names and data
+    channel_names = [channel['PnN'] for channel in fcs_data.channels]
+    data = pd.DataFrame(fcs_data.events, columns=channel_names)
+
+    return data, channel_names
 
 # Function to plot the data based on selected x and y parameters
 def plot_dot(file, x_param, y_param):
@@ -32,7 +39,12 @@ def plot_dot(file, x_param, y_param):
         plt.savefig(buf, format="png")
         buf.seek(0)
         plt.close()
-        return buf
+
+        # Save the plot as a temporary file for display
+        temp_file = "/tmp/dot_plot.png"
+        with open(temp_file, "wb") as f:
+            f.write(buf.getbuffer())
+        return temp_file
     except Exception as e:
         return f"Error: {e}"
 
@@ -59,7 +71,7 @@ with gr.Blocks() as app:
         y_param = gr.Dropdown(label="Y Parameter", choices=[], interactive=True)
 
     plot_btn = gr.Button("Generate Plot")
-    plot_output = gr.Image(type="file", label="Dot Plot")
+    plot_output = gr.Image(type="filepath", label="Dot Plot")
 
     # Update parameter dropdowns on file upload
     fcs_file.change(get_parameters, inputs=[fcs_file], outputs=[x_param, y_param])
@@ -68,4 +80,4 @@ with gr.Blocks() as app:
     plot_btn.click(plot_dot, inputs=[fcs_file, x_param, y_param], outputs=plot_output)
 
 # Run the app
-app.launch()
+app.launch()