Update src/tools/accent_tool.py

src/tools/accent_tool.py

@@ -6,6 +6,17 @@ from pydub import AudioSegment
 import whisper
 from speechbrain.pretrained.interfaces import foreign_class
 
+def clear_tmp_dir(path):
+    for filename in os.listdir(path):
+        file_path = os.path.join(path, filename)
+        try:
+            if os.path.isfile(file_path) or os.path.islink(file_path):
+                os.unlink(file_path)
+            elif os.path.isdir(file_path):
+                shutil.rmtree(file_path)
+        except Exception as e:
+            print(f'Failed to delete {file_path}. Reason: {e}')
+
 class AccentAnalyzerTool:
     def __init__(self):
         self.whisper_model = whisper.load_model("medium")
@@ -23,10 +34,10 @@ class AccentAnalyzerTool:
         try:
             self.log("Downloading video...")
             tmp_dir = "tmp"
-            # Clean up tmp folder if exists
-            if os.path.exists(tmp_dir):
-                shutil.rmtree(tmp_dir)
-            os.makedirs(tmp_dir, exist_ok=True)
+            if not os.path.exists(tmp_dir):
+                os.makedirs(tmp_dir, exist_ok=True)
+            else:
+                clear_tmp_dir(tmp_dir)
 
             video_path = os.path.join(tmp_dir, "video.mp4")
 
@@ -70,7 +81,6 @@ class AccentAnalyzerTool:
                 f"**Transcript of the audio:**\n\n *{transcript.strip(' ')}*"
             )
 
-            shutil.rmtree(tmp_dir, ignore_errors=True)
             return summary
 
         except Exception as e: