Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 from PIL import Image
+import os
 
 # Define the function signature
 def display_data(data: pd.DataFrame):
@@ -9,7 +10,7 @@ def display_data(data: pd.DataFrame):
 # Create the Streamlit app
 def main():
     st.title("Upload Files and Display Data in Function Signature")
-    
+
     # Create an upload form for CSV files
     st.subheader("Upload CSV Files")
     csv_file = st.file_uploader("Choose a CSV file", type=["csv"])
@@ -17,15 +18,25 @@ def main():
     # Create an upload form for image files
     st.subheader("Upload Image Files")
     image_file = st.file_uploader("Choose an image file", type=["jpg", "jpeg", "png"])
-    
+
+    # Add OS file listing of uploaded files
+    st.subheader("Uploaded Files")
+    files = os.listdir("uploads")
+    for file in files:
+        st.markdown(f"[{file}]({os.path.join('uploads', file)})")
+
     # Check if files are uploaded and display data in function signature
     if csv_file is not None:
-        data = pd.read_csv(csv_file)
+        with open(os.path.join("uploads", csv_file.name), "wb") as f:
+            f.write(csv_file.getbuffer())
+        data = pd.read_csv(os.path.join("uploads", csv_file.name))
         display_data(data)
-    
+
     if image_file is not None:
-        image = Image.open(image_file)
+        with open(os.path.join("uploads", image_file.name), "wb") as f:
+            f.write(image_file.getbuffer())
+        image = Image.open(os.path.join("uploads", image_file.name))
         st.image(image, caption="Uploaded Image")
-        
+
 if __name__ == "__main__":
     main()