Update app.py

app.py

@@ -8,8 +8,14 @@ import json
 import re
 
 
+def get_image_data(fig):
+    buf = io.BytesIO()
+    fig.savefig(buf, format='png')
+    buf.seek(0)
+    img = Image.open(buf)
+    return img
 
-# 执行 Python 代码并生成图像
+# Execute Python code and generate images
 def execute_code(code):
     namespace = {}
     exec(code, namespace)