app.py

import gradio as gr
import zipfile
import datetime


def add_image(camera, gallery):
    if camera is None:
        gr.Warning("No Camera Recording!")
        return
    if gallery is None:
        return [camera]
    else:
        images = gallery
        images.append(camera)
        return images


def zip_gallery(gallery):
    current_time = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
    new_zip_name = f'{current_time}.zip'
    count = 0
    if not gallery:
        return gr.DownloadButton("Download", visible=False)
    zf = zipfile.ZipFile(new_zip_name, mode="w")
    try:
        for path, _ in gallery:
            zf.write(path, f'{current_time}_{count}.png',
                     compress_type=zipfile.ZIP_DEFLATED)
            count = count + 1
    except FileNotFoundError:
        print("Error Occurred!")
        gr.Warning("Zip Error!")
    finally:
        zf.close()
        gr.Info(f"Zip Complete. Name: {new_zip_name}")

    return gr.DownloadButton("Download/下載", value=new_zip_name, visible=True)


with gr.Blocks(fill_height=True) as demo:
    with gr.Accordion("How to Use/使用說明"):
        gr.Markdown(
            """
            # How to Use
            
            1. Open Camera on left side, then click "Record" button on the camera screen

            2. Click "Capture" button to capture image to gallery

            3. After capture all image, click "Zip file" button to zip images

            4. Download zip file by clicking "Download" button showed below

            # 如何使用
            
            1. 點擊左側開啟攝影機，並按下"Record"按鈕讓攝影機能夠進行擷取

            2. 點擊"Capture"按鈕，影像就會被擷取到下方的欄位中

            3. 擷取影像完畢後，點擊下方的"Zip file"按鈕將影像打包並壓縮

            4. 按下隨後出現的"Download"按鈕就能下載打包好的壓縮檔
            
            """
        )
    with gr.Row():
        with gr.Column():
            camera = gr.Image(type="pil", format="png", webcam_constraints={
                              "video": {"width": 640, "height": 640}}, sources="webcam", mirror_webcam=False, scale=0)
            catch_btn = gr.Button("Capture/拍照", scale=0)
        with gr.Column():
            preview_img = gr.Image(visible=False)
            gallery = gr.Gallery(
                label="Collected images", show_label=False, elem_id="gallery", columns=[10], rows=[4], object_fit="contain", height="auto")
            zip_btn = gr.Button("Zip file/打包照片")
            download_btn = gr.DownloadButton("Download/下載", visible=False)

    catch_btn.click(add_image, [preview_img, gallery], [gallery])
    camera.stream(lambda s: s, camera, preview_img)
    zip_btn.click(zip_gallery, gallery, download_btn)


if __name__ == "__main__":
    demo.launch()