Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from docling.document_converter import DocumentConverter
 import json
+import os
 
 # 1) Initialize DocumentConverter
 converter = DocumentConverter()
@@ -20,8 +21,20 @@ def convert(file, out_format):
         elif out_format == "HTML":
             return doc.document.export_to_html()
         elif out_format == "DocTags":
-            # Export as DocTags and convert to JSON
-            doctags = doc.document.save_as_doctags()  # This is the JSON-like format
+            # Define a temporary file path
+            temp_filename = "converted_document.doctags"
+            
+            # Save the document as DocTags (JSON-like format)
+            doc.document.save_as_doctags(temp_filename)
+            
+            # Read the saved DocTags file back into memory
+            with open(temp_filename, 'r') as file:
+                doctags = json.load(file)
+            
+            # Clean up the temporary file (optional, to avoid clutter)
+            os.remove(temp_filename)
+            
+            # Return the content as a formatted JSON string
             return json.dumps(doctags, indent=4)
         else:
             return "Unsupported output format"