DataWizard9742 commited on
Commit
e64ed76
·
verified ·
1 Parent(s): 8abcf3f

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +16 -4
src/streamlit_app.py CHANGED
@@ -22,12 +22,24 @@ st.title("Telangana Minorities Residential Educational Institutions Society Anal
22
  uploaded_file = st.file_uploader("Upload your dataset (Excel or CSV)", type=["xlsx", "csv"])
23
 
24
  if uploaded_file:
25
- if uploaded_file.name.endswith(".xlsx"):
26
- df = pd.read_excel(uploaded_file)
 
 
 
 
 
 
27
  else:
28
- df = pd.read_csv(uploaded_file)
 
 
 
 
 
 
 
29
 
30
- st.success("Dataset uploaded successfully!")
31
 
32
  # Standardize column names
33
  df.columns = df.columns.str.strip().str.lower().str.replace(" ", "_")
 
22
  uploaded_file = st.file_uploader("Upload your dataset (Excel or CSV)", type=["xlsx", "csv"])
23
 
24
  if uploaded_file:
25
+ # Save uploaded file to /tmp for Hugging Face
26
+ save_path = os.path.join("/tmp", uploaded_file.name)
27
+ with open(save_path, "wb") as f:
28
+ f.write(uploaded_file.getbuffer())
29
+
30
+ # Read file based on extension
31
+ if uploaded_file.name.endswith(".csv"):
32
+ df = pd.read_csv(save_path)
33
  else:
34
+ df = pd.read_excel(save_path)
35
+
36
+ st.session_state["uploaded_df"] = df # Share dataset across pages
37
+ st.success("✅ File uploaded and saved successfully!")
38
+ st.write("Preview of dataset:")
39
+ st.dataframe(df.head())
40
+ else:
41
+ st.info("Please upload a dataset to get started.")
42
 
 
43
 
44
  # Standardize column names
45
  df.columns = df.columns.str.strip().str.lower().str.replace(" ", "_")