Function create_demo

demo_opendoc.py:357–460

Create Gradio demo interface. Returns: gr.Blocks: Gradio Blocks application instance.

()

# Define the Gradio Interface
def create_demo() -> gr.Blocks:
    """Create Gradio demo interface.

    Returns:
        gr.Blocks: Gradio Blocks application instance.
    """
    # Get example images path and download if necessary
    example_img_dir = get_example_images_path(demo_type='doc')

    # Get list of example images
    example_images = []
    if os.path.exists(example_img_dir):
        for file in os.listdir(example_img_dir):
            if file.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.gif')):
                example_images.append(os.path.join(example_img_dir, file))
        example_images = sorted(example_images)

    with gr.Blocks(css=custom_css,
                   theme=gr.themes.Soft(),
                   title='OpenDoc-0.1B Demo') as demo:
        # Header
        gr.HTML("""
        <div class="app-header">
            <h1>🚀 OpenDoc-0.1B</h1>
            <p>Ultra-Lightweight Document Parsing System with 0.1B Parameters (built by <a href="https://github.com/Topdu/OpenOCR">OCR Team</a>, <a href="https://fvl.fudan.edu.cn">FVL Lab</a>)</p>
            <p style="font-size: 0.95em; color: #888;">
                Powered by <a href="https://www.paddleocr.ai/latest/version3.x/module_usage/layout_analysis.html" target="_blank">PP-DocLayoutV2</a> for layout analysis and <a href="https://arxiv.org/pdf/2512.21095" target="_blank">UniRec-0.1B</a> for unified recognition of text, formulas, and tables
            </p>
        </div>
        <div class="quick-links">
            <a href="https://github.com/Topdu/OpenOCR" target="_blank">📖 GitHub</a>
            <a href="https://arxiv.org/pdf/2512.21095" target="_blank">📄 Paper</a>
            <a href="https://huggingface.co/topdu/unirec-0.1b" target="_blank">🤗 Model</a>
        </div>
        """)

        with gr.Row():
            with gr.Column(scale=4, elem_classes=['upload-section']):
                input_img = gr.Image(type='filepath',
                                     label='📤 Upload Document Image',
                                     elem_classes=['image-container'])

                # Add examples if available
                if example_images:
                    gr.Examples(
                        examples=example_images,
                        inputs=input_img,
                        label='📚 Example Documents'
                    )
                btn = gr.Button('🔍 Analyze Document',
                                variant='primary',
                                size='lg')
                gr.Markdown("""
                ### 💡 Tips
                - Supports Chinese and English documents
                - Best for reports, papers, magazines, and complex layouts
                - Handles text, formulas, tables, and images
                """)