Diddy gets to spruce up his look for his upcoming criminal trial … the judge just signed off on letting him go with a business casual look in court, instead of his jail jumpsuits.

According to new legal docs, obtained by TMZ, Diddy just got a signed order from the judge allowing him to receive non-prison clothing at the Metropolitan Detention Center in Brooklyn to wear at next month’s trial.

Diddy’s got permission to have up to 5 button-down shirts, 5 pairs of pants, 5 sweaters, 5 pairs of socks and 2 pairs of shoes without laces to wear to court.