Update app.py

Se ha agregado una barra de progreso a app.py.

app.py

@@ -181,10 +181,68 @@ with gr.Blocks(css=custom_css) as app:
                     elem_classes="gr-padded"
                 )
 
+    with gr.Row():
+        progress = gr.Slider(minimum=0, maximum=100, value=0, label="Progress", interactive=False)
+    
     convert_button.click(
-        fn=lambda file, format, method: convert_document_docling(file, format) if method == "llama-index-readers-docling" else convert_document_original(file, format),
+        fn=lambda file, format, method: convert_document_docling(file, format, progress) if method == "llama-index-readers-docling" else convert_document_original(file, format, progress),
         inputs=[file_input, format_input, method_input],
-        outputs=[output_text, output_metadata, download_button, download_button, status_message]
+        outputs=[output_text, output_metadata, download_button, download_button, status_message, progress]
     )
+    
+    def update_progress(current, total):
+        return (current / total) * 100
+    
+    def convert_document_docling(file, output_format, progress):
+        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"
+            }
+            progress(100)  # Set progress to 100% once complete
+            return converted_text, metadata, output_path, gr.update(visible=True), "✅ Document converted successfully!", 100
+        except Exception as e:
+            error_metadata = {"Error": str(e), "Status": "Failed"}
+            return "", error_metadata, None, gr.update(visible=False), "❌ Error during conversion", 0
+    
+    def convert_document_original(file, output_format, progress):
+        try:
+            converter = DocumentConverter()
+            result = converter.convert(file.name)
+            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"
+            output_filename = os.path.splitext(os.path.basename(file.name))[0] + file_extension
+            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": "docling"
+            }
+            progress(100)  # Set progress to 100% once complete
+            return converted_text, metadata, output_path, gr.update(visible=True), "✅ Document converted successfully!", 100
+        except Exception as e:
+            error_metadata = {"Error": str(e), "Status": "Failed"}
+            return "", error_metadata, None, gr.update(visible=False), "❌ Error during conversion", 0
 
 app.launch(debug=True, share=True)
+