I'd go with #1. #2 is possible, but it means the infection is the cause of the disease which the medicine doesn't treat. If that's what you mean, then fine.
New words, one handy idiom, and a 2-minute quiz — delivered to your inbox to keep your streak alive.