Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import openai
 
+
 def generate_plantuml(api_key, text):
     openai.api_key = api_key
     response = openai.ChatCompletion.create(
@@ -14,7 +15,7 @@ def generate_plantuml(api_key, text):
         ],
     )
     print(response)
-    return response["messages"][-1]["content"]
+    return response["choices"][0]["message"]["content"]
 
 iface = gr.Interface(
     fn=generate_plantuml,