Update app.py

app.py

@@ -12,12 +12,14 @@ import traceback
 
 def save_image(element, images):
     try:
-        if hasattr(element, 'stream'):
+        if hasattr(element, 'stream') and element.stream:
             image_data = element.stream.get_rawdata()
             image = Image.open(io.BytesIO(image_data))
             image_filename = f"extracted_image_{len(images)}.png"
             image.save(image_filename)
             images.append({"filename": image_filename})
+        else:
+            print("No stream data for image element")
     except Exception as e:
         print(f"Error extracting image: {e}")
 
@@ -45,6 +47,7 @@ def parse_pdf(pdf_file, output_format, progress=gr.Progress()):
                     if isinstance(element, LTTextBoxHorizontal):
                         text += element.get_text()
                     elif isinstance(element, (LTFigure, LTImage)):
+                        print(f"Processing element: {type(element)}")
                         save_image(element, images)
 
             with pdfplumber.open(pdf_file) as pdf:
@@ -115,4 +118,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()  # Temporarily disable sharing for debugging
+    iface.launch()  # Temporarily disable sharing for debugging