test/gdrive_test.py

import gradio as gr
from utils.gdrive_downloader import PublicFolderAudioLoader
from config import conf

LOADER  = PublicFolderAudioLoader(conf.GDRIVE_API_KEY)

def fetch_audio(folder_link, filename):
    sr, wav = LOADER.load_audio(folder_link, filename)
    return (sr, wav)

demo = gr.Interface(
    fn=fetch_audio,
    inputs=[
        gr.Textbox(label="Folder URL or ID",
                   value=conf.GDRIVE_FOLDER),
        gr.Textbox(label="Filename (e.g. 0001.wav)")
    ],
    outputs=gr.Audio(label="🔊 Audio"),
)

if __name__ == "__main__":
    demo.launch()