Update app.py

app.py

@@ -36,7 +36,8 @@ def GenerateImageByCode(client, message, code_prompt):
             code = data['step_details']['tool_calls'][0]['code_interpreter']['input']
             if 'image' in data['step_details']['tool_calls'][0]['code_interpreter']['outputs'][0].keys():
                 image_id = data['step_details']['tool_calls'][0]['code_interpreter']['outputs'][0]['image']['file_id']
-                image_bytes = client.files.with_raw_response.content(image_id).content
+    assert image_id
+    image_bytes = client.files.with_raw_response.content(image_id).content
     with open(f'{image_id}.png', 'wb') as f:
         f.write(image_bytes)
     base64_image = base64.b64encode(image_bytes).decode('utf-8')