ashish-001 commited on
Commit
de5e541
·
verified ·
1 Parent(s): 28b7868

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -114,7 +114,12 @@ try:
114
  shutil.rmtree(temp_dir)
115
  os.makedirs(temp_dir, exist_ok=True)
116
  st.session_state.cleaned_temp = True
117
- collection.delete(where={"path": {"$contains": "temp_folder"}})
 
 
 
 
 
118
 
119
  st.title("Extract frames from video using text")
120
  # Upload section
 
114
  shutil.rmtree(temp_dir)
115
  os.makedirs(temp_dir, exist_ok=True)
116
  st.session_state.cleaned_temp = True
117
+ results=collection.get(include=["metadatas","ids"])
118
+ ids_to_delete=[
119
+ _id for _id,metadata in zip(results["ids"],results['metadatas']) if metadata.get("path","").startswith("temp")
120
+ ]
121
+ if ids_to_delete:
122
+ collection.delete(ids=ids_to_delete)
123
 
124
  st.title("Extract frames from video using text")
125
  # Upload section