progres report fix

app.py

@@ -186,9 +186,8 @@ def run_inference(img_path, task_type="ocr", progress=gr.Progress()):
 
     try:
         logger.info(f"--- Inference Start: {task_type} ---")
-        progress(0, desc="📦 Initializing inference engine...")
+        progress(0, desc="Initializing...")
         output = pipeline.predict(input=img_path)
-        logger.info(f"Output object type: {type(output)}")
         
         md_content = ""
         json_content = ""
@@ -198,60 +197,57 @@ def run_inference(img_path, task_type="ocr", progress=gr.Progress()):
         run_output_dir = os.path.join(OUTPUT_DIR, run_id)
         os.makedirs(run_output_dir, exist_ok=True)
 
-        logger.info(f"Inference generator ready. Starting iteration...")
-        progress(0.1, desc="🔍 Document preprocessing...")
+        logger.info(f"will iterate")
     
         for i, res in enumerate(output):
             logger.info(f"Processing segment {i+1}...")
-            # Use dynamic progress increment
-            p_val = min(0.1 + (i + 1) * 0.15, 0.95)
-            progress(p_val, desc=f"✍️ Recognizing content (segment {i+1})...")
+            progress(None, desc=f"Processing segment {i+1}...")
             
-            # Save results to unique dir
-            seg_dir = os.path.join(run_output_dir, f"seg_{i}")
-            os.makedirs(seg_dir, exist_ok=True)
-            res.save_to_json(save_path=seg_dir)
-            res.save_to_markdown(save_path=seg_dir)
+            # Save results
+            res.save_to_json(save_path=run_output_dir)
+            res.save_to_markdown(save_path=run_output_dir)
             res.print()
             
-            # Gather files specifically from this segment
-            fnames = os.listdir(seg_dir)
+            # Read back generated files
+            fnames = os.listdir(run_output_dir)
             for fname in fnames:
-                fpath = os.path.join(seg_dir, fname)
+                fpath = os.path.join(run_output_dir, fname)
                 if fname.endswith(".md"):
                     with open(fpath, 'r', encoding='utf-8') as f:
                         content = f.read()
-                        md_content += content + "\n\n"
+                        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()
-                        json_content += content + "\n\n"
+                        if content not in json_content:
+                            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)
                     vis_html += f'<div style="margin-bottom:20px; border: 2px solid #10b981; border-radius: 12px; overflow: hidden; background:white;">'
                     vis_html += f'<img src="{vis_src}" alt="Vis {i+1}" style="width:100%;">'
                     vis_html += f'</div>'
             
-            # Yield partial results to keep UI alive
-            partial_md = _escape_inequalities_in_math(md_content)
-            yield partial_md, md_content, vis_html, json_content
             logger.info(f"Finished processing segment {i+1}")
+            md_preview = _escape_inequalities_in_math(md_content)
+            yield md_preview, md_content, vis_html, json_content
 
         if not md_content:
             md_content = "⚠️ Finished but no content was recognized."
             yield md_content, md_content, "", ""
-        else:
-            final_md = _escape_inequalities_in_math(md_content)
-            progress(1.0, desc="✅ Complete")
-            yield final_md, md_content, vis_html, json_content
         
         logger.info("--- Inference Finished Successfully ---")
+        progress(1.0, desc="✅ Complete")
 
     except Exception as e:
         logger.error(f"❌ Inference Error: {e}")
         logger.error(traceback.format_exc())
         yield f"❌ Error: {str(e)}", "", "", ""
-        return
+
+    except Exception as e:
+        logger.error(f"❌ Inference Error: {e}")
+        logger.error(traceback.format_exc())
+        return f"❌ Error: {str(e)}", "", "", ""
 
 # --- UI Components ---
 
@@ -291,7 +287,7 @@ with gr.Blocks() as demo:
             with gr.Row():
                 with gr.Column(scale=5):
                     file_doc = gr.Image(label="Upload Image", type="filepath")
-                    btn_parse = gr.Button("🔍 Start Parsing", variant="primary")
+                    btn_parse = gr.Button("� Start Parsing", variant="primary")
                     with gr.Row():
                         chart_switch = gr.Checkbox(label="Chart OCR", value=True)
                         unwarp_switch = gr.Checkbox(label="Unwarping", value=False)