Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
 from PIL import Image
 import os
+import base64
 from helper import (
     custom_file_uploader, resize_image, convert_image_to_base64, post_request_and_parse_response, 
     draw_bounding_boxes_for_textract, extract_text_from_textract_blocks, ChatGPTClient
 )
-from pdf2image import convert_from_path
 import tempfile
 import shutil
 
@@ -25,25 +25,30 @@ with st.sidebar:
     
     # Display a placeholder for uploaded image
     st.warning("Please upload an image or a single-page PDF file!")
-    uploaded_file = st.file_uploader("Upload an Image or PDF", type=['PDF'], label_visibility="collapsed")
+    uploaded_file = st.file_uploader("Upload an Image or PDF", type=['TXT', 'PDF'], label_visibility="collapsed")
     
     pil_image = None
     if uploaded_file:
         # Handle PDF file
         if uploaded_file.type == "application/pdf":
-            with tempfile.NamedTemporaryFile(delete=False) as temp_pdf:
-                temp_pdf.write(uploaded_file.read())
-                temp_pdf_path = temp_pdf.name
-            
-            # Convert PDF to image
             try:
-                pages = convert_from_path(temp_pdf_path, dpi=200, poppler_path="/usr/local/bin/poppler")  # Make sure poppler is installed and path is set correctly
+                # Read PDF as bytes and convert to base64
+                pdf_bytes = uploaded_file.read()
+                pdf_base64 = base64.b64encode(pdf_bytes).decode('utf-8')
+                
+                # Convert base64 to an image object
+                with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_pdf:
+                    temp_pdf.write(base64.b64decode(pdf_base64))
+                    temp_pdf_path = temp_pdf.name
+                    
+                # Convert PDF to image using pdf2image (assumed single page)
+                pages = convert_from_path(temp_pdf_path, dpi=200)
                 if len(pages) != 1:
                     st.warning("Please upload a PDF with only one page!")
                 else:
                     pil_image = pages[0]
             except Exception as e:
-                st.error(f"Failed to convert PDF to image: {e}. Please ensure that Poppler is installed and added to your system PATH.")
+                st.error(f"Failed to convert PDF to image: {e}")
             finally:
                 # Clean up the temporary file
                 if os.path.exists(temp_pdf_path):