Update app.py

app.py

@@ -20,20 +20,22 @@ def create_portfolio_graph():
         height = bar.get_height()
         ax.text(bar.get_x() + bar.get_width() / 2, height, f'{percentage*100:.0f}%', ha='center', va='bottom')
 
-    # Save the plot as an image within the Gradio workspace
-    image_path = "/mnt/data/portfolio_chart.png"
-    plt.savefig(image_path)
+    # Convert the plot to a NumPy array
+    image_array = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
+    image_array = image_array.reshape(fig.canvas.get_width_height()[::-1] + (3,))
+
+    # Close the plot
     plt.close()
 
-    # Return the image path
-    return image_path
+    # Return the image array
+    return image_array
 
 # Gradio Interface with Button for user interaction
 iface = gr.Interface(
     fn=create_portfolio_graph,
-    inputs=None,
+    inputs=gr.Button("Display Graph"),
     outputs=gr.Image(label="Portfolio Composition Chart"),
-    live=False
+    live=True
 )
 
 # Launch Gradio interface
@@ -50,3 +52,4 @@ iface.launch()
 
 
 
+