Update app.py

app.py

@@ -1,52 +1,44 @@
 import gradio as gr
 import pandas as pd
-from rpy2.robjects import pandas2ri, r
-from rpy2.robjects.packages import importr
-
-# enable pandas <-> R dataframe conversion
-pandas2ri.activate()
-
-# load R package
-eulerr = importr("eulerr")
+import matplotlib.pyplot as plt
+from matplotlib_venn import venn2, venn3
 
 def make_venn(file):
     try:
-        # read CSV
+        # Load CSV
         df = pd.read_csv(file.name)
         
-        # check minimum cols
-        if df.shape[1] < 2:
-            return "CSV must contain: primary key + up to 3 category columns."
-        
-        # drop primary key
+        # Drop primary key (first column)
         categories = df.columns[1:]
+        if len(categories) < 2:
+            return "Error: Need at least 2 category columns!"
         if len(categories) > 3:
-            return "Error: Only up to 3 category columns allowed!"
+            return "Error: Only up to 3 categories supported!"
         
-        # convert to R dataframe
-        rdf = pandas2ri.py2rpy(df[categories])
+        # Convert each category column into a set of IDs
+        sets = [set(df[df[col] == 1].index) for col in categories]
         
-        # compute Euler diagram
-        fit = eulerr.euler(rdf)
+        # Plot Venn
+        plt.figure(figsize=(6,6))
+        if len(sets) == 2:
+            venn2(sets, set_labels=categories)
+        elif len(sets) == 3:
+            venn3(sets, set_labels=categories)
         
-        # save plot
         out_file = "venn.png"
-        r(f"png('{out_file}', width=600, height=600)")
-        r("plot")(fit, fills=["#66c2a5", "#fc8d62", "#8da0cb"])
-        r("dev.off")()
-        
+        plt.savefig(out_file)
+        plt.close()
         return out_file
     
     except Exception as e:
         return f"Error: {e}"
 
-# gradio interface
 demo = gr.Interface(
     fn=make_venn,
     inputs=gr.File(file_types=[".csv"], label="Upload CSV"),
-    outputs=gr.Image(type="filepath", label="Euler Diagram"),
-    title="EulerR Venn Diagram Generator",
-    description="Upload a CSV with a primary key column + up to 3 category columns. Generates an Euler diagram."
+    outputs=gr.Image(type="filepath", label="Venn Diagram"),
+    title="Venn Diagram Generator",
+    description="Upload a CSV with a primary key column + up to 3 category columns (binary flags: 0/1)."
 )
 
 if __name__ == "__main__":