Update app.py

app.py

@@ -187,6 +187,38 @@ def uploaded_file(filename):
     logging.info(f"Serving file: {filename}")
     return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
 
+@app.route('/clear_folders', methods=['DELETE'])
+def clear_folders():
+    try:
+        # Function to clear all files in a given folder
+        def clear_folder(folder_path):
+            if os.path.exists(folder_path):
+                for filename in os.listdir(folder_path):
+                    file_path = os.path.join(folder_path, filename)
+                    try:
+                        if os.path.isfile(file_path):
+                            os.remove(file_path)
+                            logging.info(f"Deleted file: {file_path}")
+                        else:
+                            logging.warning(f"{file_path} is not a file, skipping.")
+                    except Exception as e:
+                        logging.error(f"Error deleting file {file_path}: {e}")
+                        return jsonify({'message': f"Error deleting file {file_path}"}), 500
+            else:
+                logging.warning(f"Folder {folder_path} does not exist.")
+                return jsonify({'message': f"Folder {folder_path} does not exist"}), 404
+
+        # Clear both the upload and result folders
+        clear_folder(app.config['UPLOAD_FOLDER'])
+        clear_folder(app.config['RESULT_FOLDER'])
+
+        logging.info("Both upload and result folders cleared successfully.")
+        return jsonify({'message': 'Both upload and result folders cleared successfully'}), 200
+
+    except Exception as e:
+        logging.error(f"An unexpected error occurred while clearing folders: {e}")
+        return jsonify({'message': 'Failed to clear folders'}), 500
+
 if __name__ == '__main__':
     logging.info("Starting Flask app")
     app.run(debug=True)