Update app.py

app.py

@@ -12,20 +12,13 @@ def pdf_page_to_image(pdf_path, page_number):
     if page_number < 0 or page_number >= len(doc):
         raise gr.Error("Page number out of range")
 
-    # Select the specified page
     page = doc.load_page(page_number)
-
-    # Render the page to an image
     pix = page.get_pixmap()
-
-    # Convert the image to a PIL Image
     img_data = pix.tobytes("png")  # Get the PNG image data
     img = Image.open(io.BytesIO(img_data))
-
-    # Close the PDF document
     doc.close()
 
-    return img
+    return img, {"example": "value"}
 
 def get_pdf_length(pdf_path):
     return len(fitz.open(pdf_path))
@@ -43,11 +36,11 @@ with gr.Blocks() as demo:
         
         with gr.Column():
             i = gr.Image()
-            o = gr.Textbox(label="Prediction")
+            o = gr.JSON(label="Prediction")
             e = gr.Textbox(label="Edited prediction")
         
     f.upload(update_number, f, n)
-    n.submit(pdf_page_to_image, [f, n], i)
-    b.click(pdf_page_to_image, [f, n], i)
+    n.submit(pdf_page_to_image, [f, n], [i, o])
+    b.click(pdf_page_to_image, [f, n], [i, o])
     
 demo.launch()