Update app.py

app.py

@@ -1,7 +1,10 @@
-import gradio as gr
+from flask import Flask, send_file
 import matplotlib.pyplot as plt
 import numpy as np
 
+app = Flask(__name__)
+
+@app.route('/')
 def create_portfolio_graph():
     # Data for the bars
     positions = ['Long', 'Short']
@@ -21,24 +24,15 @@ def create_portfolio_graph():
         ax.text(bar.get_x() + bar.get_width() / 2, height, f'{percentage*100:.0f}%', ha='center', va='bottom')
 
     # Save the plot as an image
-    plt.savefig("/mnt/data/portfolio_chart.png")
+    image_path = "static/portfolio_chart.png"
+    plt.savefig(image_path)
     plt.close()
 
-    # Return the image as a NumPy array
-    return np.array(plt.imread("/mnt/data/portfolio_chart.png"))
-
-# Gradio Interface (with a dummy input)
-iface = gr.Interface(
-    fn=create_portfolio_graph,
-    inputs=[gr.Textbox()],
-    outputs=gr.Image(type="numpy", label="Portfolio Composition Chart"),
-    live=True,
-    capture_session=True
-)
-
-# Launch Gradio interface
-iface.launch()
+    # Return the image path
+    return send_file(image_path, mimetype='image/png')
 
+if __name__ == '__main__':
+    app.run(debug=True)