other inference

app.py

@@ -1,4 +1,5 @@
 import atexit
+import json
 import functools
 from queue import Queue
 from threading import Event, Thread
@@ -51,15 +52,32 @@ class PaddleOCRModelManager(object):
             args, kwargs, result_queue = item
             try:
                 result = model.predict(*args, **kwargs)
-                # markdown_list = []
-                # #markdown_images = []
-                # for res in result:
-                #     md_info = res.markdown
-                #     markdown_list.append(md_info)
-                #     #markdown_images.append(md_info.get("markdown_images", {}))
-                # markdown_texts = model.concatenate_markdown_pages(markdown_list)
+                markdown_list = []
+                #markdown_images = []
+                for res in result:
+                    md_info = res.markdown
+                    markdown_list.append(md_info)
+                    #markdown_images.append(md_info.get("markdown_images", {}))
+                markdown_texts = model.concatenate_markdown_pages(markdown_list)
+
+                def serialize_for_json(obj):
+                    if isinstance(obj, dict):
+                        return {k: serialize_for_json(v) for k, v in obj.items()}
+                    elif isinstance(obj, list):
+                        return [serialize_for_json(item) for item in obj]
+                    elif hasattr(obj, '__dict__'):
+                        return serialize_for_json(obj.__dict__)
+                    elif isinstance(obj, (str, int, float, bool, type(None))):
+                        return obj
+                    else:
+                        return str(type(obj))
                 
-                result_queue.put((True, result))
+                serialized_result = serialize_for_json(result)
+                json_str = json.dumps(serialized_result, indent=4)
+
+                returns = [markdown_texts, json_str]
+                
+                result_queue.put((True, returns))
             except Exception as e:
                 result_queue.put((False, e))
             finally:
@@ -115,29 +133,7 @@ def inference(img):
         return "Please upload an image."
     
     try:
-        result = model_manager.infer(img)
-
-        
-        # if not result or len(result) == 0:
-        #     return "No text detected in the image."
-        
-        # # Serialize to JSON first (this worked before)
-        import json
-        
-        def serialize_for_json(obj):
-            if isinstance(obj, dict):
-                return {k: serialize_for_json(v) for k, v in obj.items()}
-            elif isinstance(obj, list):
-                return [serialize_for_json(item) for item in obj]
-            elif hasattr(obj, '__dict__'):
-                return serialize_for_json(obj.__dict__)
-            elif isinstance(obj, (str, int, float, bool, type(None))):
-                return obj
-            else:
-                return str(type(obj))
-        
-        serialized_result = serialize_for_json(result)
-        return json.dumps(serialized_result, indent=4)
+        return model_manager.infer(img)
         
         # # Now extract text from the serialized structure
         # extracted_texts = []