Hola... bueno te podria aconsejar respecto a la parte grafica pq con sockets solo he trabajado en lenguaje C y nunca en java....
Pues lo perfecto es crear una clase que extienda a la JPanel, es decir q utilice el codigo de esta y ademas añades metodos y atributos q necesites
Lo bueno de utilizar JPanel es que puedes añadirle lo que kieras... botones, etiquetas, lo q sea... incluso mas JPanel
Despues en el main q utilices... la clase q ejecutara la aplicacion... creas un JFrame y le añades la clase q exitiende a la JPanel... es decir la clase q utilices, de esta forma podras crear tus ventanitas de forma sencilla....
De todas formas te aconsejo q mires algun manual o libro de java... el libor q te aconsejo JAVA 2