para-lost commited on
Commit
a1a71bc
·
1 Parent(s): 91fbc34
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ with open('agent_no_image_new_lib.prompt', 'r') as f:
45
 
46
  import glob
47
  import time
48
-
49
  def list_files_in_directory(directory):
50
  """Lists all files in the given directory."""
51
  return set(glob.glob(os.path.join(directory, "*")))
@@ -55,6 +55,7 @@ def cleanup_temp_files(temp_files):
55
  for temp_file in temp_files:
56
  try:
57
  if os.path.exists(temp_file):
 
58
  os.remove(temp_file)
59
  print(f"Deleted temporary file: {temp_file}")
60
  except Exception as e:
 
45
 
46
  import glob
47
  import time
48
+ import shutil
49
  def list_files_in_directory(directory):
50
  """Lists all files in the given directory."""
51
  return set(glob.glob(os.path.join(directory, "*")))
 
55
  for temp_file in temp_files:
56
  try:
57
  if os.path.exists(temp_file):
58
+ shutil.rmtree(temp_file)
59
  os.remove(temp_file)
60
  print(f"Deleted temporary file: {temp_file}")
61
  except Exception as e:
requirements.txt CHANGED
@@ -7,5 +7,5 @@ colormath
7
  scipy
8
  matplotlib
9
  chromedriver_py
10
- bing_image_downloader
11
  transformers
 
7
  scipy
8
  matplotlib
9
  chromedriver_py
10
+ bing-image-downloader==1.1.2
11
  transformers