predict_page.py

predict_page.py

@@ -0,0 +1,38 @@
+import streamlit as st
+import os
+
+def save_uploaded_files(files, folder_path):
+    # Create the folder if it doesn't exist
+    if not os.path.exists(folder_path):
+        os.makedirs(folder_path)
+
+    saved_file_paths = []
+    for file in files:
+        # Save each uploaded file to the folder
+        file_path = os.path.join(folder_path, file.name)
+        with open(file_path, "wb") as f:
+            f.write(file.getbuffer())
+        saved_file_paths.append(file_path)
+
+    return saved_file_paths
+
+# Streamlit app
+def show_predict_page():
+    st.title("Multi-image Uploader")
+    
+    # File uploader
+    uploaded_files = st.file_uploader("Upload four images", type=["jpg", "jpeg", "png"], accept_multiple_files=True, key="file_uploader")
+    
+    if uploaded_files is not None:
+        # Save the uploaded files to a folder
+        folder_path = "image_folder"
+        saved_file_paths = save_uploaded_files(uploaded_files, folder_path)
+        
+        # Display the uploaded images
+        for file_path in saved_file_paths:
+            st.image(file_path, caption="Uploaded Image", use_column_width=True)
+        
+        # Display the file paths
+        st.write("Files saved at:")
+        for file_path in saved_file_paths:
+            st.write(file_path)