Update app.py

app.py

@@ -1,36 +1,27 @@
 import streamlit as st
-# import transformers
+# import torch
 # from transformers import AutoProcessor, UdopForConditionalGeneration
 # from datasets import load_dataset
 
 # processor = AutoProcessor.from_pretrained("microsoft/udop-large", apply_ocr=True)
 # model = UdopForConditionalGeneration.from_pretrained("microsoft/udop-large")
 
-st.title("Image Selection and Upload")
-st.write("Select or upload an image to test the model.")
+st.title("Document (/Image) Selection and Upload")
+st.write("Select or upload a document (/an image) to test the model.")
 
+# Image selection
+uploaded_files = st.file_uploader("Upload document(s) [/image(s)]:", type=["docx", "pdf", "jpg", "jpeg", "png"], accept_multiple_files=True)
+selected_file = st.selectbox("Select a document (/an image):", uploaded_files, format_func=lambda file: file.name if file else "None")
 
-# Image upload
-uploaded_file = st.file_uploader("Upload an image:", type=["jpg", "jpeg", "png"])
-if uploaded_file is not None:
-    st.image(uploaded_file, caption="Uploaded Image")
+# Display selected image
+if selected_file is not None and selected_file != "None":
+    st.image(selected_file, caption="Selected Image")
 
 # Model testing button
-if st.button("Test Model"):
-    if image_option == "Image 1":
-        # Test the model with Image 1
-        st.write("Testing the model with Image 1...")
-    elif image_option == "Image 2":
-        # Test the model with Image 2
-        st.write("Testing the model with Image 2...")
-    elif image_option == "Image 3":
-        # Test the model with Image 3
-        st.write("Testing the model with Image 3...")
-    elif uploaded_file is not None:
-        # Test the model with the uploaded image
-        st.write("Testing the model with the uploaded image...")
-    else:
-        st.write("Please select or upload an image.")
+if st.button("Test Model") and selected_file != "None":
+    st.write("Testing the model with the selected image...")
+elif st.button("Test Model") and selected_file == "None":
+    st.write("Please upload and select an image.")
 
 # encoding = processor(image, question, words, boxes=boxes, return_tensors="pt")