Wizualizacja struktury katalogu w postaci tekstowej

Mam następujące zadanie: zdefiniuj funkcję, która generuje strukturę danego katalogu w postaci tekstowej.
I mam taki kod

import os

for katalog, katalogi, pliki in os.walk('.'):
    
    for pod_katalog in katalogi:
        print(os.path.join( katalog, pod_katalog), '/')
        
    for plik in pliki:
        print(os.path.join(katalog, plik))

który działa tak, że wyświetla najpierw wszystkie podkatalogi jeden po drugim, a potem dopiero wszystkie pliki.
Nie wiem jak to zrobić żeby program wyświetlał pliki z danego podkatalogu bezpośrdnio pod jego nazwą i dopiero później przechodził dalej, kolejny podkatalog i jego pliki, podkatalog i jego pliki itd.

Rekurencja.

Nie uczyliśmy się jeszcze za bardzo pisać programów z rekurencją, mogę prosić o jakąś dokładniejszą wskazówkę?