Create app.py

app.py

@@ -0,0 +1,35 @@
+import gradio as gr
+import openai
+import matplotlib.pyplot as plt
+import io
+import numpy as np
+import base64
+from PIL import Image
+
+# 将图像转换为 base64，以便在 gradio 中显示
+def get_image_data(plt):
+    buf = io.BytesIO()
+    plt.savefig(buf, format='PNG')
+    buf.seek(0)
+    img = Image.open(buf)
+    return img
+
+# 执行 Python 代码并生成图像
+def execute_code(code):
+    exec(code)
+    return get_image_data(plt)
+
+openai.api_key = 'your-openai-key'
+
+def gpt3_inference(prompt):
+    response = openai.Completion.create(
+      engine="text-davinci-002", 
+      prompt=prompt, 
+      max_tokens=100
+    )
+    code = response.choices[0].text.strip()
+    img = execute_code(code)
+    return img
+
+iface = gr.Interface(fn=gpt3_inference, inputs="text", outputs="image")
+iface.launch()