Update app.py

app.py

@@ -143,7 +143,9 @@ async def generate_new_img(hex_code: str, prompt: str = Query(..., description="
             image_data = io.BytesIO()
             image.save(image_data, format="PNG")
             image_data.seek(0)
-            created_image = image
+
+            global generated_image
+            generated_image = image
             return StreamingResponse(image_data, media_type="image/png")
         except Exception as e:
             raise HTTPException(status_code=500, detail=str(e))