Somando datas no python

Boa noite pessoal, podem me ajudar ?

Estou aprendendo python e estou fazendo um mini projeto de biblioteca, a minha dificuldade está em fazer a soma da data de empréstimo de um livro com 7 dias de prazo para leitura , com o dia em que ele pegou o livro, tentei construir uma função para exercitar esse método, porém ele não faz a conta estou sem ideia de como arrumar isso, alguma ideia ?

from datetime import datetime, timedelta

print('---$Bem vindo a nossa Biblioteca$---')

livros = ['Crepusculo', 'diario de um banana', 'A cabana', 'A luz do amanhã']
print(livros)

print()
print('----------------------------------')

print('escolha um livro:')
usuario = input()

print('----------------------------------')
print(f'o livro escolhido foi', usuario)
print('----------------------------------')
print('você pegou o livro em: ')
dia_emprestimo = datetime.now()
print(dia_emprestimo.strftime('%d/%m/%y'))

print('----------------------------------')
print('Data da devolução: ')


def prazo():
    dia_emprestimo + timedelta(days=7)


devolucao = prazo()

print(devolucao)

A função tem que retornar o resultado (ou seja, precisa de um return):

def prazo():
    return dia_emprestimo + timedelta(days=7)
    ^^^^^^

Além disso, seria mais interessante se a função recebesse a data como parâmetro:

def prazo(data_emprestimo):
    return data_emprestimo + timedelta(days=7)

dia_emprestimo = datetime.now()
# print, etc...

devolucao = prazo(dia_emprestimo)

Assim vc pode reusá-la com qualquer data.

tentei a segunda solução e funcionou, a primeira também obrigado Hugo.