DrishtiSharma commited on
Commit
e09a0e3
Β·
verified Β·
1 Parent(s): 9cfc436

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +7 -4
src/streamlit_app.py CHANGED
@@ -33,10 +33,10 @@ persist_dir = tempfile.mkdtemp()
33
  def prepare_vectorstore(uploaded_files=None):
34
  documents = []
35
 
36
- if uploaded_files:
37
  st.write("πŸ“ Uploaded files:")
38
  for file in uploaded_files:
39
- st.write(f"β€’ {file.name}")
40
  file_path = Path(tempfile.gettempdir()) / file.name
41
  try:
42
  with open(file_path, "wb") as f:
@@ -61,9 +61,12 @@ def prepare_vectorstore(uploaded_files=None):
61
  st.error(f"❌ Error loading {file.name}:")
62
  st.exception(e)
63
  st.text(traceback.format_exc())
64
-
65
  else:
66
- st.info("No files uploaded. Loading from documents folder...")
 
 
 
 
67
  try:
68
  loader = DirectoryLoader(str(documents_path), glob="**/*.txt", loader_cls=TextLoader)
69
  documents = loader.load()
 
33
  def prepare_vectorstore(uploaded_files=None):
34
  documents = []
35
 
36
+ if uploaded_files and any(file.size > 0 for file in uploaded_files):
37
  st.write("πŸ“ Uploaded files:")
38
  for file in uploaded_files:
39
+ st.write(f"β€’ {file.name} ({file.size} bytes)")
40
  file_path = Path(tempfile.gettempdir()) / file.name
41
  try:
42
  with open(file_path, "wb") as f:
 
61
  st.error(f"❌ Error loading {file.name}:")
62
  st.exception(e)
63
  st.text(traceback.format_exc())
 
64
  else:
65
+ if uploaded_files:
66
+ st.warning("⚠️ Uploaded files detected, but they seem to be empty.")
67
+ else:
68
+ st.info("πŸ“‚ No files uploaded. Loading from documents folder...")
69
+
70
  try:
71
  loader = DirectoryLoader(str(documents_path), glob="**/*.txt", loader_cls=TextLoader)
72
  documents = loader.load()