from gradio_client import Client
import gradio as gr
import os

def check_password(username, password):
    if password == os.environ["ACCESS"]:
        return True
    else:
        return False

read_key = os.environ.get("HF_TOKEN", None)

if __name__ == "__main__":
    client = Client.duplicate("Nauryzbay/deckify_private", hf_token=read_key)

    func = lambda file, number_of_pages: client.predict(file, number_of_pages)

    with gr.Blocks() as demo:
        file = gr.File(label="Upload PDF")
        number_of_pages = gr.Number(label="Number of pages")
        output = gr.Textbox(label="Output")
        greet_btn = gr.Button("Generate slides")
        greet_btn.click(fn=func, inputs=[file, number_of_pages], outputs=output, api_name="greet")
    
    demo.queue(max_size=10)
    demo.launch(auth=check_password)