Render plot as piechart

app.py

@@ -4,6 +4,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 import io
 from PIL import Image
+import base64
 
 client = Client("https://duchaba-friendly-text-moderation.hf.space/--replicas/gffry/")
 
@@ -14,26 +15,16 @@ def moderate_text(text, safer_value):
         api_name="/censor_me"
     )
 
-    # Assuming 'result' contains moderation categories and their respective counts
-    categories = result.get("categories", {"Safe": 1, "Unsafe": 1})  # Example structure
-    labels = list(categories.keys())
-    sizes = list(categories.values())
-    
-    # Generate a pie chart
-    fig, ax = plt.subplots()
-    ax.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=90)
-    ax.axis('equal')  # Equal aspect ratio ensures that pie is drawn as a circle.
-    ax.set_title('Moderation Result Distribution')
+    # Example structure of the result
+    base64_data = result.get('plot', '').split(',')[1]
 
-    # Save plot to a bytes buffer
-    buf = io.BytesIO()
-    plt.savefig(buf, format='png')
-    buf.seek(0)
-    
-    # Convert bytes buffer to PIL Image
-    plot_image = Image.open(buf)
+    # Decode base64 to bytes
+    img_data = base64.b64decode(base64_data)
 
-    return result, plot_image
+    # Convert bytes to PIL Image
+    img = Image.open(io.BytesIO(img_data))
+
+    return result, img
 
 # Define the Gradio interface
 demo = gr.Interface(