Update app.py

app.py

@@ -131,7 +131,10 @@ def visualize_clusters(df):
     plt.title('Clusters of User Queries')
     plt.xlabel('PCA Component 1')
     plt.ylabel('PCA Component 2')
-    plt.show()
+    buf = BytesIO()
+    plt.savefig(buf, format='png')
+    buf.seek(0)
+    return buf
 
 def silhouette_analysis(X, labels, num_clusters):
     fig, ax1 = plt.subplots(1, 1)
@@ -160,7 +163,10 @@ def silhouette_analysis(X, labels, num_clusters):
     ax1.set_yticks([])
     ax1.set_xticks([i/10.0 for i in range(-1, 11)])
     
-    return fig
+    buf = BytesIO()
+    plt.savefig(buf, format='png')
+    buf.seek(0)
+    return buf
 
 def main(file, num_clusters_to_display):
     try:
@@ -171,7 +177,8 @@ def main(file, num_clusters_to_display):
         
         df = preprocess_data(df)
         df, X, kmeans = cluster_data(df, num_clusters=15)
-        visualize_clusters(df)
+        
+        cluster_plot = visualize_clusters(df)
 
         cluster_sizes = df['Cluster'].value_counts()
         sorted_clusters = cluster_sizes.index.tolist()
@@ -191,9 +198,9 @@ 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, silhouette_plot
     except Exception as e:
+        print(f"Error: {e}")
         return str(e), None, None
 
 interface = gr.Interface(
@@ -205,10 +212,10 @@ interface = gr.Interface(
     outputs=[
         gr.File(label="Clustered Data CSV"),
         gr.Number(label="Silhouette Score"),
-        gr.Plot(label="Silhouette Plot")
+        gr.Image(label="Silhouette Plot")
     ],
     title="Unanswered User Queries Clustering",
     description="Unanswered User Query Categorization"
 )
 
-interface.launch(share=True)
+interface.launch(share=True)