Python's `input()` fn Implementation in Rust

input.rs

use std::fmt::Display;
use std::io::{stdin, stdout, Write};

pub fn input(prompt: impl Display) -> Box<str> {
    let mut buffer = String::new();

    stdout().write(prompt.to_string().as_bytes()).unwrap();
    stdout().flush().unwrap();
    stdin().read_line(&mut buffer).unwrap();

    buffer.trim().into()
}

fn main() {
    let output = input("Input: ");
    println!("Output: {}", output);
}