process_pdf

Added process_pdf

app.py

@@ -267,7 +267,12 @@ def apply_ocr(cell_coordinates, cropped_table):
     return df, data
 
 
-def process_pdf(image):
+def process_pdf(pdf, image):
+    # if pdf:
+    #    image = convert_from_bytes()
+    return process_image(image)
+
+def process_image(image):
     cropped_table = detect_and_crop_table(image)
 
     image, cells = recognize_table(cropped_table)
@@ -289,7 +294,7 @@ description = """<h3>Sherif's Demo for table extraction with the Table Transform
 """
 
 app = gr.Interface(fn=process_pdf, 
-                     inputs=gr.Image(type="pil"), 
+                     inputs=[gr.Interface(file_types=['.pdf'], label="Upload a PDF file"), gr.Image(type="pil", label="Or upload an image file to perform OCR directly")],
                      outputs=[gr.Image(type="pil", label="Detected table"), gr.Dataframe(label="Table as CSV"), gr.JSON(label="Data as JSON")],
                      title=title,
                      description=description)