Update app.py

app.py

@@ -9,6 +9,7 @@ import re
 from io import BytesIO
 import tempfile
 import numpy as np
+from PIL import Image
 
 def preprocess_data(df):
     df.rename(columns={'Question Asked': 'texts'}, inplace=True)
@@ -134,7 +135,8 @@ def visualize_clusters(df):
     buf = BytesIO()
     plt.savefig(buf, format='png')
     buf.seek(0)
-    return buf
+    img = Image.open(buf)
+    return img
 
 def silhouette_analysis(X, labels, num_clusters):
     fig, ax1 = plt.subplots(1, 1)
@@ -166,7 +168,8 @@ def silhouette_analysis(X, labels, num_clusters):
     buf = BytesIO()
     plt.savefig(buf, format='png')
     buf.seek(0)
-    return buf
+    img = Image.open(buf)
+    return img
 
 def main(file, num_clusters_to_display):
     try:
@@ -198,10 +201,10 @@ def main(file, num_clusters_to_display):
 
         with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmpfile:
             df.to_csv(tmpfile.name, index=False)
-            return tmpfile.name, silhouette_avg, silhouette_plot
+            return tmpfile.name, silhouette_avg, cluster_plot, silhouette_plot
     except Exception as e:
         print(f"Error: {e}")
-        return str(e), None, None
+        return str(e), None, None, None
 
 interface = gr.Interface(
     fn=main,
@@ -212,6 +215,7 @@ interface = gr.Interface(
     outputs=[
         gr.File(label="Clustered Data CSV"),
         gr.Number(label="Silhouette Score"),
+        gr.Image(label="Cluster Plot"),
         gr.Image(label="Silhouette Plot")
     ],
     title="Unanswered User Queries Clustering",