Update app.py

app.py

@@ -21,7 +21,12 @@ def codeinterpreter(openai_key,prompt, files):
     #async with CodeInterpreterSession(model="gpt-3.5-turbo",openai_api_key="") as session:
         response = session.generate_response_sync(prompt, fileList,True)
 
-        return [response.content,response.files]
+        images = []
+
+        for _file in response.files:
+            images.append(_file.get_image())
+
+        return [response.content,images,response.files]
         
 
 
@@ -30,6 +35,6 @@ app = gr.Interface(
         gr.Textbox(label="openai_key"),
         gr.Textbox(label="prompt",info="input the prompt"), 
         gr.Files(),
-    ], outputs=["text","files"]
+    ], outputs=["text","gallery","files"]
 )
 app.launch()