If you're lucky enough to spend one day more in Tokyo, go and see "jingu baseball park firework fes", near Shinjuku, on August 1, too. This is the festival I like the best in Tokyo.
I don't know if you can get the ticket, rather difficult from overseas, but I am sure you will fully enjoy the firework outside the park.