feat: Add a JSON output tab to display raw inference results for both Document and Visual Language tasks.

app.py

@@ -219,15 +219,16 @@ def run_inference(img_path, task_type="ocr", progress=gr.Progress()):
                 logger.info(f"Processing file {fname}...")
                 fpath = os.path.join(run_output_dir, fname)
                 if fname.endswith(".md"):
+                    logger.info(f"Processing MD file {fname}...")
                     with open(fpath, 'r', encoding='utf-8') as f:
                         content = f.read()
+                        logger.info(f"MD content: {content}")
                         if content not in md_content:
                             md_content += content + "\n\n"
                 elif fname.endswith(".json"):
                     with open(fpath, 'r', encoding='utf-8') as f:
                         content = f.read()
-                        if content not in json_content:
-                            json_content += content + "\n\n"
+                        json_content += content + "\n\n"
                 elif fname.endswith((".png", ".jpg", ".jpeg")) and ("res" in fname or "vis" in fname):
                     vis_src = image_to_base64_data_url(fpath)
                     new_vis = f'<div style="margin-bottom:20px; border: 2px solid #10b981; border-radius: 12px; overflow: hidden; background:white;">'
@@ -303,20 +304,22 @@ with gr.Blocks() as demo:
                             vis_image_doc = gr.HTML('<div style="text-align:center; color:#94a3b8; padding: 50px;">Results will appear here.</div>')
                         with gr.Tab("📜 Raw Source"):
                             md_raw_doc = gr.Code(language="markdown")
+                        with gr.Tab("💾 JSON Feed"):
+                            json_doc = gr.Code(language="json")
 
             def parse_doc_wrapper(fp, ch, uw, progress=gr.Progress()):
                 if not fp:
-                    yield "⚠️ Please upload an image.", "", ""
+                    yield "⚠️ Please upload an image.", "", "", ""
                     return
                 # Initial yield to force loading indicators on all tabs
-                yield "⌛ Initializing...", gr.update(value="<p>⌛ Processing...</p>"), "⌛ Initializing..."
-                for res_preview, res_raw, res_vis, _ in run_inference(fp, task_type="Document", progress=progress):
-                    yield res_preview, res_vis, res_raw
+                yield "⌛ Initializing...", gr.update(value="<p>⌛ Processing...</p>"), "⌛ Initializing...", "{}"
+                for res_preview, res_raw, res_vis, res_json in run_inference(fp, task_type="Document", progress=progress):
+                    yield res_preview, res_vis, res_raw, res_json
 
             btn_parse.click(
                 parse_doc_wrapper, 
                 [file_doc, chart_switch, unwarp_switch], 
-                [md_preview_doc, vis_image_doc, md_raw_doc],
+                [md_preview_doc, vis_image_doc, md_raw_doc, json_doc],
                 show_progress="full"
             )
 
@@ -336,19 +339,22 @@ with gr.Blocks() as demo:
                             md_preview_vl = gr.Markdown(latex_delimiters=LATEX_DELIMS, elem_classes="output-box")
                         with gr.Tab("📜 Source"):
                             md_raw_vl = gr.Code(language="markdown")
+                        with gr.Tab("💾 JSON Feed"):
+                            json_vl = gr.Code(language="json")
 
             def run_vl_wrapper(fp, prompt, progress=gr.Progress()):
                 if not fp:
-                    yield "⚠️ Please upload an image.", ""
+                    yield "⚠️ Please upload an image.", "", ""
                     return
-                for res_preview, res_raw, _, _ in run_inference(fp, task_type=prompt, progress=progress):
-                    yield res_preview, res_raw
+                yield "⌛ Initializing...", "⌛ Initializing...", "{}"
+                for res_preview, res_raw, _, res_json in run_inference(fp, task_type=prompt, progress=progress):
+                    yield res_preview, res_raw, res_json
 
             for btn, prompt in [(btn_ocr, "Text"), (btn_formula, "Formula"), (btn_table, "Table")]:
                 btn.click(
                     run_vl_wrapper, 
                     [file_vl, gr.State(prompt)], 
-                    [md_preview_vl, md_raw_vl],
+                    [md_preview_vl, md_raw_vl, json_vl],
                     show_progress="full"
                 )