Those of us who play by ear and can also read music on a pinch
will alter a song so that we can under stand the piece better.
I hope that the OP will not alter the Bach flute partita, though.
There has been enough abuse of Bach's compositions already.
Tomato said that there was a piece for a flute like you want.
No, I said that the composition which he already had in his possession was written for unaccompanied flute, so he already had everything he needed.
That is if you are able to compose and alter it to fit the piano. Try it.
As I understood it, his goal was not to perform the composition with flute and piano, but rather to perform the composition as Bach intended it.
At least, I hope that was his goal.