|
|
import gradio as gr
|
|
|
from helper_fns import process_files, get_summarization_method
|
|
|
from summarizer import summarize_files
|
|
|
|
|
|
with gr.Blocks() as demo:
|
|
|
with gr.Row():
|
|
|
with gr.Column():
|
|
|
files = gr.UploadButton(
|
|
|
label='Upload Files For Summarization',
|
|
|
file_count='multiple',
|
|
|
file_types=["pdf", "docx", "pptx"]
|
|
|
)
|
|
|
summarization_method_radio = gr.Radio(choices=['map_reduce', 'stuff', 'refine'],
|
|
|
value='map_reduce',
|
|
|
label='Select Summarization Method',
|
|
|
interactive=False)
|
|
|
|
|
|
generate_summaries_button = gr.Button(value='Generate Summaries',
|
|
|
interactive=False,
|
|
|
elem_id='summary_button')
|
|
|
|
|
|
|
|
|
|
|
|
files.upload(process_files, None, outputs=[generate_summaries_button,
|
|
|
summarization_method_radio])
|
|
|
summarization_method_radio.input(fn = get_summarization_method,
|
|
|
inputs=summarization_method_radio)
|
|
|
|
|
|
with gr.Column():
|
|
|
summary_text = gr.Textbox(label='Summarized Text: ',
|
|
|
interactive=False)
|
|
|
|
|
|
|
|
|
generate_summaries_button.click(
|
|
|
fn = summarize_files,
|
|
|
inputs=[summarization_method_radio, files],
|
|
|
outputs=[summary_text]
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
