Update app.py

app.py

@@ -6,7 +6,7 @@ import numpy as np
 import base64
 from PIL import Image
 
-# 将图像转换为 base64，以便在 gradio 中显示
+# Convert image to base64 for display in gradio
 def get_image_data(plt):
     buf = io.BytesIO()
     plt.savefig(buf, format='PNG')
@@ -14,16 +14,15 @@ def get_image_data(plt):
     img = Image.open(buf)
     return img
 
-# 执行 Python 代码并生成图像
+# Execute Python code and generate images
 def execute_code(code):
     exec(code)
     return get_image_data(plt)
 
-openai.api_key = 'sk-tDHORTwNMfpIJGeeHNRpFx2CaiEKeqsoJH2cfncNyl67DWoU'
-
-def gpt3_inference(prompt):
+def gpt_inference(prompt, model, openai_key):
+    openai.api_key = openai_key
     response = openai.Completion.create(
-      engine="gpt-3.5-turbo", 
+      engine=model, 
       prompt=prompt, 
       max_tokens=100
     )
@@ -31,5 +30,10 @@ def gpt3_inference(prompt):
     img = execute_code(code)
     return img
 
-iface = gr.Interface(fn=gpt3_inference, inputs="text", outputs="image")
-iface.launch()
+iface = gr.Interface(
+    fn=gpt_inference, 
+    inputs=["text", gr.inputs.Dropdown(choices=["gpt3.5-turbo", "gpt4"], label="Model"), "text"], 
+    outputs=gr.outputs.Image(type="pil"),
+    input_labels=["Prompt", "Model", "OpenAI Key"]
+)
+iface.launch()