path change for img

src/file_manage.py

@@ -1,33 +1,21 @@
-import streamlit as st 
 import os
-
-SAVE_DIR = 'uploads'
-os.makedirs(SAVE_DIR, exist_ok=True)
+import streamlit as st
 
 def upload_csv():
-    uploaded_file = st.file_uploader("Choose a CSV file",type="csv")
-    
+    uploaded_file = st.file_uploader("Choose a CSV file", type="csv")
     if uploaded_file is not None:
-        # Save the uploaded file to a specific directory
-        save_path = os.path.join('uploads', uploaded_file.name)
-        with open(save_path, 'wb') as f:
+        # Save to /tmp for Hugging Face Spaces compatibility
+        save_path = os.path.join("/tmp", uploaded_file.name)
+        with open(save_path, "wb") as f:
             f.write(uploaded_file.getbuffer())
-        st.success(f"File {uploaded_file.name} uploaded successfully!")
-        
+        st.success(f"File saved to {save_path}")
+
 def list_files():
-    st.subheader("List of Uploaded Files")
-    files = os.listdir('uploads')
+    # List files in /tmp directory
+    files = [f for f in os.listdir("/tmp") if f.endswith(".csv")]
     if files:
+        st.write("Uploaded CSV files:")
         for file in files:
-            col1, col2 = st.columns([10, 1])
-            col1.write(file)
-            if col2.button("Delete", key=file):
-                os.remove(os.path.join('uploads',file))
-                st.success(f'File {file} deleted successfully!')
-                st.rerun()
+            st.write(file)
     else:
-        st.info("No files uploaded yet.")
-    
-    return files
-
-   
+        st.write("No CSV files found in /tmp.")