JARVISXIRONMAN commited on
Commit
8cd9ba5
·
verified ·
1 Parent(s): e319081

Create components/upload_files.py

Browse files
Files changed (1) hide show
  1. components/upload_files.py +17 -12
components/upload_files.py CHANGED
@@ -1,18 +1,23 @@
1
- # upload_files.py
2
 
3
- import streamlit as st
4
  import os
 
 
 
5
 
6
- def run_upload_files():
7
- st.title("📤 Upload Business Documents")
8
- uploaded_file = st.file_uploader("Upload your business PDF", type=["pdf"])
9
 
10
- if uploaded_file:
11
- save_path = os.path.join("user_docs", uploaded_file.name)
12
- os.makedirs("user_docs", exist_ok=True)
13
 
14
- with open(save_path, "wb") as f:
15
- f.write(uploaded_file.read())
16
 
17
- st.session_state.pdf_path = save_path # Save for use in Q&A
18
- st.success("✅ File uploaded and ready to use in PDF Q&A.")
 
 
 
 
 
 
 
1
+ # components/upload_files.py
2
 
 
3
  import os
4
+ import shutil
5
+ import streamlit as st
6
+ from utils.session import save_to_json
7
 
8
+ UPLOAD_DIR = "uploaded_docs"
9
+ os.makedirs(UPLOAD_DIR, exist_ok=True)
 
10
 
11
+ def run_upload_files():
12
+ st.subheader("📤 Upload Business Document")
 
13
 
14
+ uploaded_file = st.file_uploader("Choose a PDF file", type=["pdf"])
 
15
 
16
+ if uploaded_file is not None:
17
+ file_path = os.path.join(UPLOAD_DIR, uploaded_file.name)
18
+
19
+ with open(file_path, "wb") as f:
20
+ shutil.copyfileobj(uploaded_file, f)
21
+
22
+ save_to_json("pdf_path", file_path)
23
+ st.success(f"✅ File '{uploaded_file.name}' uploaded successfully and ready for Q&A.")