Update app.py

app.py

@@ -9,13 +9,17 @@ import streamlit as st
 st.title("CIC Demo (by ITT)")
 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)
+# File selection
+uploaded_files = st.file_uploader("Upload document(s) [/image(s)]:", type=["docx", "pdf", "pptx", "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")
 
-# Display selected image
+# Display selected file
 if selected_file is not None and selected_file != "None":
-    st.image(selected_file, caption="Selected Image")
+    file_extension = selected_file.name.split(".")[-1]
+    if file_extension in ["jpg", "jpeg", "png"]:
+        st.image(selected_file, caption="Selected Image")
+    else:
+        st.write("Selected file: ", selected_file.name)
 
 # Model testing button
 testButton = st.button("Test Model")