Tengo rojo la cabeza del pene

Hace varios meses tuve una relación sexual desprotegida y un mes después empecé a sentirme raro y fui al medico me diagnóstico con epididimitis después regrese por los síntomas no desaparecían fue allí que me ise unos estudios y me dio un par de antibióticos y una injection ala semana después se me enrojeció el pene y no es fácil retraerlo estuve buscan en internet y pare como si fuera balanitis pero no estoy seguro ayuda es algo muy estresante!