Scideck / app.py
Nauryzbay's picture
changed upload button
8da6c61
raw
history blame
1.21 kB
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)
def upload_file(file):
return file.name
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_output = gr.File()
upload_button = gr.UploadButton("Click to Upload a PDF File", file_types=["file"], file_count="single")
upload_button.upload(upload_file, upload_button, file_output)
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=[upload_button, number_of_pages, secret], outputs=output, api_name="greet")
demo.queue(max_size=5)
demo.launch()