Update app.py

app.py

@@ -44,22 +44,30 @@ hf_writer_incorrect = gr.HuggingFaceDatasetSaver(
 #             return function(*args, **kwargs)
 
 
-def display_file(input_file):
-    global current_file_path
-    current_file_path = input_file.name if input_file else None
-    if not input_file:
+def display_file(input_files):
+    global current_file_paths
+
+    # Initialize the list of current file paths
+    current_file_paths = [file.name for file in input_files]
+
+    if not input_files:
         return gr.HTML.update(visible=False), gr.Image.update(visible=False)
-    if input_file.name.endswith(".pdf"):
-        with open(input_file.name, "rb") as input_file:
-            pdf_base64 = base64.b64encode(input_file.read()).decode()
-        return gr.HTML.update(
-            PDF_IFRAME.format(pdf_base64), visible=True
-        ), gr.Image.update(visible=False)
+    
+    # Check if there's any PDF file among the uploaded files
+    pdf_base64 = None
+    for input_file in input_files:
+        if input_file.name.endswith(".pdf"):
+            with open(input_file.name, "rb") as pdf_file:
+                pdf_base64 = base64.b64encode(pdf_file.read()).decode()
+                break  # Assuming only one PDF is present
+
+    if pdf_base64:
+        return gr.HTML.update(PDF_IFRAME.format(pdf_base64), visible=True), gr.Image.update(visible=False)
     else:
-        # image = Image.open(input_file.name)
-        return gr.HTML.update(visible=False), gr.Image.update(
-            input_file.name, visible=True
-        )
+        # You can choose to display the first image in the list or handle multiple images differently
+        image = Image.open(input_files[0].name)
+        return gr.HTML.update(visible=False), gr.Image.update(image, visible=True)
+
 
 
 def show_intermediate_outputs(show_intermediate):