Update app.py

app.py

@@ -1,36 +1,58 @@
 import streamlit as st
 from docling.parsers import DOCXParser, PDFParser, PPTXParser
 from docling.utils import to_json, to_markdown
+import sys
+import traceback
 
-st.title("Docling Document Processor")
+def init_parsers():
+    parsers = {
+        "pdf": PDFParser(),
+        "docx": DOCXParser(),
+        "pptx": PPTXParser()
+    }
+    return parsers
 
-# File uploader
-uploaded_file = st.file_uploader("Choose a document (PDF, DOCX, PPTX)", type=["pdf", "docx", "pptx"])
-
-if uploaded_file is not None:
-    # Determine file type and parse
-    file_extension = uploaded_file.name.split(".")[-1].lower()
-    if file_extension == "pdf":
-        parser = PDFParser()
-    elif file_extension == "docx":
-        parser = DOCXParser()
-    elif file_extension == "pptx":
-        parser = PPTXParser()
-    else:
-        st.error("Unsupported file type.")
-
-    # Parse the document
+def process_document(file, parser):
     try:
-        document = parser.parse(uploaded_file)
+        document = parser.parse(file)
+        return document, None
     except Exception as e:
-        st.error(f"Error parsing document: {e}")
+        error_msg = f"Error: {str(e)}\n{traceback.format_exc()}"
+        return None, error_msg
+
+def main():
+    st.set_page_config(page_title="Document Processor")
+    st.title("Docling Document Processor")
+    
+    try:
+        parsers = init_parsers()
+    except ImportError as e:
+        st.error("Error: Required packages not installed. Please run: pip install docling python-docx pdfminer.six python-pptx")
+        return
 
-    # Display output options
-    output_format = st.radio("Select output format:", ("Markdown", "JSON"))
+    uploaded_file = st.file_uploader("Choose a document", type=list(parsers.keys()))
+    
+    if uploaded_file:
+        file_extension = uploaded_file.name.split(".")[-1].lower()
+        parser = parsers.get(file_extension)
+        
+        document, error = process_document(uploaded_file, parser)
+        
+        if error:
+            st.error(f"Failed to parse document:\n{error}")
+            return
+            
+        output_format = st.radio("Select output format:", ("Markdown", "JSON"))
+        
+        try:
+            if output_format == "Markdown":
+                st.subheader("Markdown Output:")
+                st.markdown(to_markdown(document))
+            else:
+                st.subheader("JSON Output:")
+                st.json(to_json(document))
+        except Exception as e:
+            st.error(f"Error converting document: {str(e)}")
 
-    if output_format == "Markdown":
-        st.subheader("Markdown Output:")
-        st.write(to_markdown(document))
-    else:
-        st.subheader("JSON Output:")
-        st.json(to_json(document))
+if __name__ == "__main__":
+    main()