add reset route

README.md

@@ -11,6 +11,7 @@ The `main` branch contains a template for a container that will spin up an API t
 3. `/upload-audio/` : Upload audio files, save to noisy audio directory
 4. `/enhance/` : Initialize model, enhance audio files, save to enhanced audio directory
 5. `/download-enhanced/` : Download enhanced audio files
+6. `/reset/` : Remove all existing audio files for another batch of enhancement
 
 To add your own model to this template, there are a few things that a miner must do:
 

app/app.py

@@ -78,6 +78,7 @@ class ModelAPI:
         self.app.post("/upload-audio/")(self.upload_audio)
         self.app.post("/enhance/")(self.enhance_audio)
         self.app.get("/download-enhanced/")(self.download_enhanced)
+        self.app.get("/reset/")(self.reset)
         
     def get_status(self):
         try:
@@ -152,6 +153,22 @@ class ModelAPI:
         except Exception as e:
             # Log the error if needed, and raise an HTTPException to inform the client
             raise fastapi.HTTPException(status_code=500, detail=f"An error occurred while creating the download file: {str(e)}")
+        
+    def reset(self):
+        """
+        Removes all audio files in preparation for another batch of enhancement.
+        """
+        for directory in [self.noisy_audio_path, self.enhanced_audio_path]:
+            if not os.path.isdir(directory):
+                continue
+
+            for filename in os.listdir(directory):
+                filepath = os.path.join(directory, filename)
+                if os.path.isfile(filepath):
+                    try:
+                        os.remove(filepath)
+                    except Exception as e:
+                        print(f"Error removing {filepath}: {e}")
 
     def run(self):