Update app.py

He actualizado el archivo app.py para incluir la funcionalidad de llama-index-readers-docling como un método de conversión adicional.

app.py

@@ -1,71 +1,60 @@
-
 import gradio as gr
 from docling.document_converter import DocumentConverter
+from llama_index.readers.docling import DoclingReader
 import json
 import tempfile
 import os
 
-def convert_document(file, output_format):
-    """
-    Converts a document to Markdown or JSON format using Docling.
-    Args:
-        file: Uploaded file to convert.
-        output_format: Desired output format (Markdown or JSON).
-    Returns:
-        Tuple containing the converted text, metadata, and downloadable file.
-    """
+def convert_document_docling(file, output_format):
+    try:
+        reader = DoclingReader(export_type=DoclingReader.ExportType.JSON if output_format == "JSON" else DoclingReader.ExportType.MARKDOWN)
+        docs = reader.load_data(file_path=file.name)
+        converted_text = docs[0].text
+        temp_dir = tempfile.gettempdir()
+        output_filename = os.path.splitext(os.path.basename(file.name))[0] + (".json" if output_format == "JSON" else ".md")
+        output_path = os.path.join(temp_dir, output_filename)
+        with open(output_path, 'w', encoding='utf-8') as f:
+            f.write(converted_text)
+        metadata = {
+            "Filename": file.name,
+            "File Size": f"{os.path.getsize(file.name) / 1024:.2f} KB",
+            "Output Format": output_format,
+            "Conversion Status": "Success",
+            "Method": "llama-index-readers-docling"
+        }
+        return converted_text, metadata, output_path, gr.update(visible=True), "✅ Document converted successfully!"
+    except Exception as e:
+        error_metadata = {"Error": str(e), "Status": "Failed"}
+        return "", error_metadata, None, gr.update(visible=False), "❌ Error during conversion"
+
+def convert_document_original(file, output_format):
     try:
-        # Initialize the converter and load the document
         converter = DocumentConverter()
         result = converter.convert(file.name)
-        
-        # Create temporary file for download
         temp_dir = tempfile.gettempdir()
-        
         if output_format == "Markdown":
             converted_text = result.document.export_to_markdown()
             file_extension = ".md"
         else:
             converted_text = result.document.export_to_json()
             file_extension = ".json"
-            
-        # Create output file
         output_filename = os.path.splitext(os.path.basename(file.name))[0] + file_extension
         output_path = os.path.join(temp_dir, output_filename)
-        
-        # Write content to file
         with open(output_path, 'w', encoding='utf-8') as f:
             f.write(converted_text)
-            
         metadata = {
             "Filename": file.name,
             "File Size": f"{os.path.getsize(file.name) / 1024:.2f} KB",
             "Output Format": output_format,
-            "Conversion Status": "Success"
+            "Conversion Status": "Success",
+            "Method": "docling"
         }
-        
-        return (
-            converted_text,
-            metadata,
-            output_path,
-            gr.update(visible=True),
-            "✅ Document converted successfully!"
-        )
-        
+        return converted_text, metadata, output_path, gr.update(visible=True), "✅ Document converted successfully!"
     except Exception as e:
-        error_metadata = {
-            "Error": str(e),
-            "Status": "Failed"
-        }
-        return (
-            "",
-            error_metadata,
-            None,
-            gr.update(visible=False),
-            "❌ Error during conversion"
-        )
-
-# Custom CSS
+        error_metadata = {"Error": str(e), "Status": "Failed"}
+        return "", error_metadata, None, gr.update(visible=False), "❌ Error during conversion"
+
+
 custom_css = """
 :root {
     --primary-color: #2563eb;
@@ -132,7 +121,6 @@ body {
 }
 """
 
-# Create Gradio interface
 with gr.Blocks(css=custom_css) as app:
     gr.HTML(
         """
@@ -142,7 +130,6 @@ with gr.Blocks(css=custom_css) as app:
         </div>
         """
     )
-    
     with gr.Row():
         with gr.Column(scale=1):
             with gr.Group():
@@ -158,19 +145,23 @@ with gr.Blocks(css=custom_css) as app:
                     value="Markdown",
                     elem_classes="gr-input"
                 )
+                method_input = gr.Radio(
+                    choices=["docling", "llama-index-readers-docling"],
+                    label="Conversion Method",
+                    value="docling",
+                    elem_classes="gr-input"
+                )
                 convert_button = gr.Button(
                     "🔄 Convert Document",
                     variant="primary",
                     elem_classes=["gr-button"]
                 )
-                
             status_message = gr.Textbox(
                 label="Status",
                 interactive=False,
                 visible=False,
                 elem_classes="gr-padded"
             )
-    
         with gr.Column(scale=2):
             with gr.Group():
                 gr.Markdown("### Conversion Output")
@@ -189,19 +180,11 @@ with gr.Blocks(css=custom_css) as app:
                     visible=False,
                     elem_classes="gr-padded"
                 )
-    
-    # Event handlers
+
     convert_button.click(
-        fn=convert_document,
-        inputs=[file_input, format_input],
-        outputs=[
-            output_text,
-            output_metadata,
-            download_button,
-            download_button,
-            status_message
-        ]
+        fn=lambda file, format, method: convert_document_docling(file, format) if method == "llama-index-readers-docling" else convert_document_original(file, format),
+        inputs=[file_input, format_input, method_input],
+        outputs=[output_text, output_metadata, download_button, download_button, status_message]
     )
 
-# Launch the app with share=True
-app.launch(debug=True, share=True)
+app.launch(debug=True, share=True)