| 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 | |
| def func(file, number_of_pages, secret): | |
| if secret != os.environ["ACCESS"]: | |
| return "Wrong password, please try again" | |
| else: | |
| client.predict(file, number_of_pages) | |
| read_key = os.environ.get("HF_TOKEN", None) | |
| if __name__ == "__main__": | |
| client = Client.duplicate("Nauryzbay/deckify_private", hf_token=read_key) | |
| with gr.Blocks() as demo: | |
| file = gr.File(label="Upload PDF") | |
| number_of_pages = gr.Number(label="Number of pages") | |
| secret = gr.Textbox(label="Password", type="password") | |
| output = gr.Textbox(label="Output") | |
| greet_btn = gr.Button("Generate slides") | |
| greet_btn.click(fn=func, inputs=[file, number_of_pages, secret], outputs=output, api_name="greet") | |
| demo.queue(max_size=5) | |
| demo.launch() |