CbCR_to_Excel / app.py
ADucatez's picture
Reorganisation of files
25e496d
import gradio as gr
import os
import shutil
import tempfile
from reduce_and_convert_PDF import reduce_and_convert
def clear_gradio_temp(exclude_files):
temp_dir = tempfile.gettempdir()
for folder in os.listdir(temp_dir):
folder_path = os.path.join(temp_dir, folder)
if "gradio" in folder_path.lower():
should_skip = any(file_path.startswith(folder_path) for file_path in exclude_files)
if should_skip:
continue
try:
shutil.rmtree(folder_path)
print(f"Deleted: {folder_path}")
except Exception as e:
print(f"Failed to delete {folder_path}: {e}")
def ui(input_files):
clear_gradio_temp(input_files)
output_zip = "./output.zip"
if os.path.exists(output_zip):
os.remove(output_zip)
input_folder = "./input_folder"
if os.path.exists(input_folder):
shutil.rmtree(input_folder)
os.makedirs(input_folder)
# Move files into the extract_folder
for file_path in input_files:
print(file_path)
shutil.copy(file_path, input_folder)
reduce_and_convert(input_folder)
return output_zip
with gr.Blocks() as appli:
gr.Markdown("## CBCR PDF to Excel Conversion Tool")
input_files = gr.File(label="Select an input folder", file_count="directory")
process_button = gr.Button("Process Files")
download_link = gr.File(label="Processed Zip")
process_button.click(fn=ui, inputs=input_files, outputs=download_link)
appli.launch()