Update app.py

app.py

@@ -29,7 +29,7 @@ from io import BytesIO
 from pdf2image import convert_from_bytes, convert_from_path     # pip install pdf2image
 
 import requests
-from utils import convert_json_to_markdown
+from utils import convert_json_to_markdown, extract_json_content
 
 def setup_poppler_linux():
     poppler_dir = "/tmp/poppler"
@@ -255,11 +255,18 @@ async def doc_parser(doc_path, prompt, model_id):
         async for chunk in request(query, model_name, client, Authorization):
             pages += chunk
             yield extract_makrdown(pages), pages
-        all_pages.append(extract_makrdown(pages))
-        
-        pages = convert_json_to_markdown(pages)
-        all_pages_raw.append(pages)
-        print(all_pages)
+
+        try:
+            json_pages = extract_json_content(pages)
+            json_pages = json.dumps(json.loads(json_pages), indent==4, ensure_ascii=False)
+        except Exception as e:
+            json_pages = pages
+        all_pages.append(extract_makrdown(json_pages))
+        try:
+            markdown_pages = convert_json_to_markdown(pages)
+        except Exception as e:
+            markdown_pages = pages
+        all_pages_raw.append(markdown_pages)
         yield "\n---\n".join(all_pages), "\n\n".join(all_pages_raw)