Update app.py

app.py

@@ -70,19 +70,28 @@ def process_pdf(pdf_file):
 
 # Create Gradio interface
 
-iface = gr.Interface(
-    fn=process_pdf,
-    inputs=gr.File(label="Upload a PDF"),
-    outputs=[
-            
-                gr.Gallery(label="Figures from PDF", object_fit='scale-down'),
-                gr.Gallery(label="Tables from PDF", object_fit='scale-down')
-    ],
-    title="PDF figures and tables extraction based on YOLOv10",
-    description="Upload a PDF file to get figures and tables in the PDF"
-)
+with gr.Blocks() as app:
+    gr.Markdown(
+        """
+        # PDF Figures and Tables Extraction
+        Upload a PDF file to extract figures and tables using YOLOv10.
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column():
+            file_input = gr.File(label="Upload a PDF")
+        with gr.Column():
+            extract_button = gr.Button("Extract")
+    
+    with gr.Row():
+        with gr.Column():
+            figures_gallery = gr.Gallery(label="Figures from PDF", object_fit='scale-down')
+        with gr.Column():
+            tables_gallery = gr.Gallery(label="Tables from PDF", object_fit='scale-down')
+    
+    extract_button.click(process_pdf, inputs=file_input, outputs=[figures_gallery, tables_gallery])
 
-# Launch the app
-iface.launch()
+app.launch()